Searched refs:begin_dims (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_slice_op.cc | 169 memory::dims begin_dims; member
172 MklSliceParams(const memory* from, const memory* to, memory::dims begin_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()
288 key_creator.AddAsKey(sliceParams.begin_dims); in CreateKey()
366 memory::dims begin_dims = {}; in ComputeMklSlice() local
367 begin_dims.resize(begin.size()); in ComputeMklSlice()
368 for (size_t i = 0; i < begin.size(); ++i) begin_dims[i] = begin[i]; in ComputeMklSlice()
398 begin_dims = MklDnnDimsInNCHW(begin_dims, input_tf_format); in ComputeMklSlice()
427 begin_dims, size_dims); in ComputeMklSlice()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2325 nvinfer1::Dims begin_dims, size_dims, stride_dims; in ConvertStridedSliceHelper() local
2326 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(begin, &begin_dims, in ConvertStridedSliceHelper()
2335 *const_cast<nvinfer1::ITensor*>(input.tensor()), begin_dims, size_dims, in ConvertStridedSliceHelper()
|