Searched refs:SetDim (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 65 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()
288 SetDim(spatial_size(), DimIndex::Y, value);
292 SetDim(spatial_size(), DimIndex::X, value);
296 SetDim(spatial_size(), dim, value);
408 SetDim(input_filter_dims(), DimIndex::Y, value);
412 SetDim(input_filter_dims(), DimIndex::X, value);
420 SetDim(input_filter_dims(), dim, value);
522 SetDim(padding(), DimIndex::Y, value);
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 144 SetDim(i, value); in RuntimeShape()
186 inline void SetDim(int i, int32 val) { in SetDim() function
282 SetDim(i, pad_value); in RuntimeShape()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 141 SetDim(i, value); in RuntimeShape()
179 inline void SetDim(int i, int32 val) { in SetDim() function
266 SetDim(i, pad_value); in RuntimeShape()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | transpose_test.cc | 59 output_shape.SetDim(i, input_shape.Dims(perms[i])); in RunTestPermutation()
|