Home
last modified time | relevance | path

Searched refs:SetDim (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dtranspose_utils.cc73 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()
Dtypes.h152 SetDim(i, value); in RuntimeShape()
194 inline void SetDim(int i, int32 val) { in SetDim() function
288 SetDim(i, pad_value); in RuntimeShape()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h65 inline void SetDim(absl::Span<int64> data, DimIndex dim, int64 value) { in SetDim() function
69 inline void SetDim(std::vector<int64>* data, DimIndex dim, int64 value) { in SetDim() function
70 return SetDim(absl::MakeSpan(*data), dim, value); in SetDim()
303 SetDim(spatial_size(), DimIndex::Y, value);
307 SetDim(spatial_size(), DimIndex::X, value);
311 SetDim(spatial_size(), dim, value);
423 SetDim(input_filter_dims(), DimIndex::Y, value);
427 SetDim(input_filter_dims(), DimIndex::X, value);
435 SetDim(input_filter_dims(), dim, value);
537 SetDim(padding(), DimIndex::Y, value);
[all …]
/external/tensorflow/tensorflow/lite/experimental/delegates/hexagon/builders/tests/
Dtranspose_test.cc73 output_shape.SetDim(i, input_shape.Dims(perms[i])); in ComputeExpectedTransposeResult()
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_test.cc47 output_shape.SetDim(i, input_shape.Dims(perms[i])); in RunTestPermutation()