Home
last modified time | relevance | path

Searched refs:dimension_index (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc154 const int dimension_index = num_splits_and_index.index(); in HandleTileShardedInputs() local
160 auto result = CreateSplitOp(num_splits, dimension_index, location, in HandleTileShardedInputs()
175 CreateSplitOp(num_splits, dimension_index, location, in HandleTileShardedInputs()
478 const auto dimension_index = dimension_and_output_splits.index(); in ValidateAndGetTiledExecuteOutputShape() local
482 if (output_shape[dimension_index] == mlir::ShapedType::kDynamicSize) { in ValidateAndGetTiledExecuteOutputShape()
488 cluster_func_output_type.getShape()[dimension_index]; in ValidateAndGetTiledExecuteOutputShape()
496 dimension_index, output_splits, output_shape_at_dim)); in ValidateAndGetTiledExecuteOutputShape()
499 new_output_shape[dimension_index] = in ValidateAndGetTiledExecuteOutputShape()
500 output_shape[dimension_index] / output_splits; in ValidateAndGetTiledExecuteOutputShape()
/external/tensorflow/tensorflow/lite/kernels/
Dmirror_pad.cc119 int64_t left_pad = 0, right_pad = 0, dimension_index, index_in_input; in GetFlatIndex() local
133 dimension_index = index / (*eval_data->output_dims_num_elements)[i]; in GetFlatIndex()
135 GetInputDimension(dimension_index, left_pad, right_pad, in GetFlatIndex()
/external/tensorflow/tensorflow/dtensor/mlir/ir/
Dtf_dtensor.cc70 const int dimension_index = dim_and_index.index(); in verify() local
81 dimension_index, dim, num_shard_for_dim)); in verify()
/external/deqp/external/openglcts/modules/gles31/
Des31cArrayOfArraysTests.cpp1730 size_t dimension_index = 0; in test_shader_compilation() local
1775 for (dimension_index = 0; dimension_index < API::MAX_ARRAY_DIMENSIONS - 1; dimension_index++) in test_shader_compilation()
1777 shader_source += variable_basenames[dimension_index]; in test_shader_compilation()
1781 shader_source += variable_basenames[dimension_index]; in test_shader_compilation()
1811 for (dimension_index = 0; dimension_index < API::MAX_ARRAY_DIMENSIONS - 1; dimension_index++) in test_shader_compilation()
1813 shader_source += variable_basenames[dimension_index]; in test_shader_compilation()
1817 shader_source += variable_basenames[dimension_index]; in test_shader_compilation()
1989 size_t dimension_index, in recursively_initialise() argument
1994 if (dimension_index == 0) in recursively_initialise()
2002 …for (size_t indent_index = dimension_index; indent_index < (API::MAX_ARRAY_DIMENSIONS + 1); indent… in recursively_initialise()
[all …]
Des31cArrayOfArraysTests.hpp712 …std::string recursively_initialise(std::string var_type, size_t dimension_index, std::string init_…
961 …std::string recursively_initialise(std::string var_type, size_t dimension_index, std::string init_…
1080 …std::string recursively_initialise(std::string var_type, size_t dimension_index, std::string init_…
1319 …std::string recursively_initialise(std::string var_type, size_t dimension_index, std::string init_…
/external/tensorflow/tensorflow/compiler/xla/service/
Dalgebraic_simplifier.cc3734 int64_t dimension_index = 0; in HandlePad() local
3743 padding_dimensions.push_back(dimension_index); in HandlePad()
3745 padding_dimensions.push_back(dimension_index); in HandlePad()
3747 dimension_index++; in HandlePad()
3774 if (pad_dims < dimension_index && in HandlePad()
3806 while (--dimension_index > j) { in HandlePad()
3807 broadcast_shape1.DeleteDimension(dimension_index); in HandlePad()
3808 pad_shape1.DeleteDimension(dimension_index); in HandlePad()
3811 while (--dimension_index >= 0) { in HandlePad()
3812 broadcast_shape1.DeleteDimension(dimension_index); in HandlePad()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emitter.h355 int64_t dimension_index);
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc1768 const TfLiteTensor& output_tensor, int dimension_index, int node_index, in CheckTensorsDimensionMatch() argument
1770 if (SizeOfDimension(&input_tensor, dimension_index) != in CheckTensorsDimensionMatch()
1771 SizeOfDimension(&output_tensor, dimension_index)) { in CheckTensorsDimensionMatch()
1776 dimension_index, SizeOfDimension(&input_tensor, dimension_index), in CheckTensorsDimensionMatch()
1777 SizeOfDimension(&output_tensor, dimension_index), op_name, in CheckTensorsDimensionMatch()