Home
last modified time | relevance | path

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

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