Searched refs:size_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 168 memory::dims size_dims; member
171 memory::dims size_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()
296 key_creator.AddAsKey(sliceParams.size_dims); in CreateKey()
377 memory::dims size_dims = {}; in ComputeMklSlice() local
379 size_dims.resize(size.size()); in ComputeMklSlice()
381 size_dims[i] = size[i]; in ComputeMklSlice()
382 if (size_dims[i] == 0) empty = true; in ComputeMklSlice()
391 auto shape_to = MklDnnDimsToTFShape(size_dims); in ComputeMklSlice()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2879 nvinfer1::Dims begin_dims, size_dims, stride_dims; in ConvertStridedSliceHelper() local
2882 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(size, &size_dims, in ConvertStridedSliceHelper()
2889 *input.tensor(), begin_dims, size_dims, stride_dims); in ConvertStridedSliceHelper()
|