Home
last modified time | relevance | path

Searched defs:rdim (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dsymbolic_shapes.cc84 const auto& rdim = right.dim(i); in ShapesSymbolicallyEqual() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/lhlo/transforms/
H A Dlhlo_legalize_to_parallel_loops.cc236 for (const auto& rdim : reduceOp.getDimensions().getValues<APInt>()) { in createReduceOpInNestedParallelLoops() local
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
H A Dgrpc_session_test.cc350 Node* rdim = test::graph::Constant(&graph, rdim_tensor); in FindMaxEigen() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dalgebraic_simplifier.cc4525 for (auto rdim : reverse->dimensions()) { in TryToReorderSliceAndReverse() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dir_emitter.cpp5058 int64_t rdim = -1; in emitIntAndSliceIndexing() local