Searched defs:dimension_index (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | xla_sharding_util.cc | 154 const int dimension_index = num_splits_and_index.index(); in HandleTileShardedInputs() local
478 const auto dimension_index = dimension_and_output_splits.index(); in ValidateAndGetTiledExecuteOutputShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ir/ |
| H A D | tf_dtensor.cc | 70 const int dimension_index = dim_and_index.index(); in verify() local
|
| /aosp_15_r20/external/deqp/external/openglcts/modules/gles31/ |
| H A D | es31cArrayOfArraysTests.cpp | 1752 size_t dimension_index = 0; in test_shader_compilation() local
2005 … size_t dimension_index, in recursively_initialise()
2570 … size_t dimension_index, in recursively_initialise()
2763 …g ExpressionsAssignment1<API>::recursively_initialise(std::string var_type, size_t dimension_index, in recursively_initialise()
3298 …xpressionsIndexingArray2<API>::recursively_initialise(std::string var_type, size_t dimension_index, in recursively_initialise()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | mirror_pad.cc | 119 int64_t left_pad = 0, right_pad = 0, dimension_index, index_in_input; in GetFlatIndex() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| H A D | xnnpack_delegate.cc | 1768 const TfLiteTensor& output_tensor, int dimension_index, int node_index, in CheckTensorsDimensionMatch()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | algebraic_simplifier.cc | 3734 int64_t dimension_index = 0; in HandlePad() local
|