Searched refs:tensor_stride (Results 1 – 4 of 4) sorted by relevance
294 const Tensor& tensor_stride = context->input(4); in Compute() local295 auto value_stride = tensor_stride.flat<int32>(); in Compute()296 stride.resize(tensor_stride.shape().num_elements()); in Compute()404 const Tensor& tensor_stride = context->input(4); in Compute() local405 auto value_stride = tensor_stride.flat<int32>(); in Compute()406 stride.resize(tensor_stride.shape().num_elements()); in Compute()510 const Tensor& tensor_stride = context->input(4); in Compute() local511 auto value_stride = tensor_stride.flat<int32>(); in Compute()512 stride.resize(tensor_stride.shape().num_elements()); in Compute()716 const Tensor& tensor_stride = context->input(4); in Compute() local[all …]
379 const Tensor& tensor_stride = context->input(2);380 auto value_stride = tensor_stride.flat<int32>();381 stride.resize(tensor_stride.shape().num_elements());
72 int tensor_stride, const Array& sub_array, in CopyArrayToSubArray() argument82 is_bias ? 1 : tensor_stride, start_idx1, start_idx2, in CopyArrayToSubArray()
102 int tensor_stride, const Array& sub_array,