| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | transpose_utils.cc | 73 input_shape->SetDim(0, 1); in RemoveOneSizeDimensions()
75 output_shape->SetDim(0, 1); in RemoveOneSizeDimensions()
87 input_shape->SetDim(new_dims_cnt, input_shape->Dims(i)); in RemoveOneSizeDimensions()
100 output_shape->SetDim(new_dims_cnt, output_shape->Dims(i)); in RemoveOneSizeDimensions()
144 non_flatten_input_shape->SetDim(i - skip_dims_cnt, input_shape.Dims(i)); in Flatten()
145 non_flatten_output_shape->SetDim(i - skip_dims_cnt, output_shape.Dims(i)); in Flatten()
|
| D | types.h | 171 SetDim(i, value); in RuntimeShape()
218 inline void SetDim(int i, int32_t val) { in SetDim() function
312 SetDim(i, pad_value); in RuntimeShape()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | batch_to_space_nd.h | 32 new_shape.SetDim(0, shape.Dims(0)); in ExtendShapeBatchToSpace()
33 new_shape.SetDim(1, shape.Dims(1)); in ExtendShapeBatchToSpace()
34 new_shape.SetDim(3, shape.Dims(2)); in ExtendShapeBatchToSpace()
|
| D | space_to_batch_nd.h | 33 new_shape.SetDim(0, shape.Dims(0)); in ExtendShapeSpaceToBatch()
34 new_shape.SetDim(1, shape.Dims(1)); in ExtendShapeSpaceToBatch()
35 new_shape.SetDim(3, shape.Dims(2)); in ExtendShapeSpaceToBatch()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 66 inline void SetDim(absl::Span<int64> data, DimIndex dim, int64 value) { in SetDim() function
70 inline void SetDim(std::vector<int64>* data, DimIndex dim, int64 value) { in SetDim() function
71 return SetDim(absl::MakeSpan(*data), dim, value); in SetDim()
280 SetDim(spatial_size(), DimIndex::Y, value); in set_height()
284 SetDim(spatial_size(), DimIndex::X, value); in set_width()
288 SetDim(spatial_size(), dim, value); in set_spatial_dim()
400 SetDim(input_filter_dims(), DimIndex::Y, value); in set_input_filter_height()
404 SetDim(input_filter_dims(), DimIndex::X, value); in set_input_filter_width()
412 SetDim(input_filter_dims(), dim, value); in set_spatial_dim()
514 SetDim(padding(), DimIndex::Y, value); in set_zero_padding_height()
[all …]
|
| /external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/ |
| D | ragged_tensor_to_tensor_tflite.cc | 127 result.SetDim(i, GetTensorData<int32_t>(&tensor)[i]); in TensorShapeFromTensor()
132 result.SetDim(i, GetTensorData<int64_t>(&tensor)[i]); in TensorShapeFromTensor()
238 result.SetDim(i, -1); in CombineRaggedTensorToTensorShapes()
244 result.SetDim(need_to_set + i, value_shape.Dims(i)); in CombineRaggedTensorToTensorShapes()
266 result_shape.SetDim(0, first_dimension); in CalculateOutputSize()
270 result_shape.SetDim(i, in CalculateOutputSize()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | batch_matmul.cc | 402 transposed_shape.SetDim(rank - 1, shape.Dims(rank - 2)); in TransposeRowsColumnsImpl()
403 transposed_shape.SetDim(rank - 2, shape.Dims(rank - 1)); in TransposeRowsColumnsImpl()
435 swapped_shape.SetDim(dims - 2, shape.Dims(dims - 1)); in SwapRowColumnDims()
436 swapped_shape.SetDim(dims - 1, shape.Dims(dims - 2)); in SwapRowColumnDims()
|
| D | transpose_test.cc | 54 output_shape.SetDim(i, input_shape.Dims(perms[i])); in RunTestPermutation()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
| D | transpose_test.cc | 78 output_shape.SetDim(i, input_shape.Dims(perms[i])); in ComputeExpectedTransposeResult()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 339 SetDim(i, value);
381 inline void SetDim(int i, int32_t val) {
475 SetDim(i, pad_value);
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5004 new_shape.SetDim(0, shape.Dims(0)); in BatchToSpaceND()
5005 new_shape.SetDim(1, shape.Dims(1)); in BatchToSpaceND()
5006 new_shape.SetDim(3, shape.Dims(2)); in BatchToSpaceND()
|