Home
last modified time | relevance | path

Searched defs:index_rank (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc552 int64 index_rank = hlo->operand(1)->shape().rank(); in GatherIndexSharding() local
736 const int64 index_rank = hlo.operand(1)->shape().rank(); in ScatterEffectiveIndexSharding() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc2255 int64 index_rank = c->Value(index_rank_dim); in ValidateSparseTensor() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc1977 int64_t index_rank = index_ty.getRank(); in Verify() local