Searched refs:begin_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 167 memory::dims begin_dims; member
170 MklSliceParams(const memory* from, const memory* to, memory::dims begin_dims, in MklSliceParams()
172 : from(from), to(to), begin_dims(begin_dims), size_dims(size_dims) {} in MklSliceParams()
235 sliceParams.size_dims, sliceParams.begin_dims); in Setup()
295 key_creator.AddAsKey(sliceParams.begin_dims); in CreateKey()
374 memory::dims begin_dims = {}; in ComputeMklSlice() local
375 begin_dims.resize(begin.size()); in ComputeMklSlice()
376 for (size_t i = 0; i < begin.size(); ++i) begin_dims[i] = begin[i]; in ComputeMklSlice()
409 begin_dims = is_slice2d in ComputeMklSlice()
410 ? MklDnnDimsInNCHW(begin_dims, input_tf_format) in ComputeMklSlice()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | strided_slice_test.cc | 68 TfLiteIntArray* begin_dims = IntArrayFromInts(begin_shape); in TestStridedSliceFloat() local
77 CreateTensor(begin_data, begin_dims), in TestStridedSliceFloat()
97 TfLiteIntArray* begin_dims = IntArrayFromInts(begin_shape); in TestStridedSliceQuantized() local
108 CreateTensor(begin_data, begin_dims), in TestStridedSliceQuantized()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2879 nvinfer1::Dims begin_dims, size_dims, stride_dims; in ConvertStridedSliceHelper() local
2880 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(begin, &begin_dims, in ConvertStridedSliceHelper()
2889 *input.tensor(), begin_dims, size_dims, stride_dims); in ConvertStridedSliceHelper()
|