Home
last modified time | relevance | path

Searched refs:indices_rank (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/fuzzing/
Dscatter_nd_fuzz.cc76 size_t indices_rank = 1 + (data[data_ix++] % kMaxIndicesRank); in FuzzImpl() local
79 if (data_ix + indices_rank >= size) { in FuzzImpl()
84 for (i = 0; i < indices_rank; i++) { in FuzzImpl()
105 for (i = 0; i < indices_rank - 1; i++) { in FuzzImpl()
109 int64 last = indices_dims[indices_rank - 1]; in FuzzImpl()
/external/tensorflow/tensorflow/lite/kernels/
Dgather_nd.cc63 const int indices_rank = NumDimensions(indices); in Prepare() local
64 const int indices_nd = SizeOfDimension(indices, indices_rank - 1); in Prepare()
69 if (indices_rank < 1) { in Prepare()
84 const int output_rank = indices_rank + params_rank - indices_nd - 1; in Prepare()
87 for (int i = 0; i < indices_rank - 1; ++i) { in Prepare()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dgather_op.cc133 int64 indices_rank = in XlaGather() local
135 dim_numbers.add_offset_dims(i + indices_rank - num_index_dims); in XlaGather()
/external/tensorflow/tensorflow/core/ops/
Ddata_flow_ops.cc109 const int64 indices_rank = c->Rank(indices_shape); in DynamicStitchShapeFunction() local
118 TF_RETURN_IF_ERROR(c->Subshape(data_shape, indices_rank, &rest)); in DynamicStitchShapeFunction()