Searched refs:size_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_slice_op.cc | 170 memory::dims size_dims; member
173 memory::dims size_dims) in MklSliceParams()
174 : from(from), to(to), begin_dims(begin_dims), size_dims(size_dims) {} in MklSliceParams()
231 src_pd, sliceParams.size_dims, sliceParams.begin_dims)); in Setup()
289 key_creator.AddAsKey(sliceParams.size_dims); in CreateKey()
369 memory::dims size_dims = {}; in ComputeMklSlice() local
371 size_dims.resize(size.size()); in ComputeMklSlice()
373 size_dims[i] = size[i]; in ComputeMklSlice()
374 if (size_dims[i] == 0) empty = true; in ComputeMklSlice()
383 auto shape_to = MklDnnDimsToTFShape(size_dims); in ComputeMklSlice()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2325 nvinfer1::Dims begin_dims, size_dims, stride_dims; in ConvertStridedSliceHelper() local
2328 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(size, &size_dims, in ConvertStridedSliceHelper()
2335 *const_cast<nvinfer1::ITensor*>(input.tensor()), begin_dims, size_dims, in ConvertStridedSliceHelper()
|