Home
last modified time | relevance | path

Searched defs:original_indices (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/cc/gradients/
Ddata_flow_grad.cc81 auto original_indices = Reshape( in DynamicPartitionGrad() local
/external/tensorflow/tensorflow/core/tfrt/graph_executor/
Dgraph_executor.cc320 std::vector<int>& original_indices) { in CreateSortedNamesAndOriginalIndices()
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dmeta_spmd_expander.cc1070 mlir::Value original_indices = one_hot_op.indices(); in ExpandOp() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc502 std::vector<int64_t> original_indices(indices.begin(), indices.end()); in ReverseSharding() local