Home
last modified time | relevance | path

Searched refs:output_shape_rank (Results 1 – 3 of 3) sorted by relevance

/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_tensor_to_tensor_tflite.cc235 const int output_shape_rank = ragged_rank + value_shape.DimensionsCount(); in CombineRaggedTensorToTensorShapes() local
236 result.Resize(output_shape_rank); in CombineRaggedTensorToTensorShapes()
237 for (int i = 0; i < output_shape_rank; ++i) { in CombineRaggedTensorToTensorShapes()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc505 int64 output_shape_rank = in InsertDegenerateDims() local
508 CHECK_LT(dim_to_insert, output_shape_rank); in InsertDegenerateDims()
512 output_shape_dim_bounds.reserve(output_shape_rank); in InsertDegenerateDims()
515 for (int64 i = 0; i < output_shape_rank; ++i) { in InsertDegenerateDims()
Dshape_inference.cc3228 const int64 output_shape_rank = in ValidateGatherDimensionNumbers() local
3233 if (offset_dim < 0 || offset_dim >= output_shape_rank) { in ValidateGatherDimensionNumbers()
3238 i, offset_dim, output_shape_rank); in ValidateGatherDimensionNumbers()