Searched refs:tensor_stride (Results 1 – 4 of 4) sorted by relevance
278 const Tensor& tensor_stride = context->input(4); in Compute() local279 auto value_stride = tensor_stride.flat<int32>(); in Compute()280 stride.resize(tensor_stride.shape().num_elements()); in Compute()402 const Tensor& tensor_stride = context->input(4); in Compute() local403 auto value_stride = tensor_stride.flat<int32>(); in Compute()404 stride.resize(tensor_stride.shape().num_elements()); in Compute()507 const Tensor& tensor_stride = context->input(4); in Compute() local508 auto value_stride = tensor_stride.flat<int32>(); in Compute()509 stride.resize(tensor_stride.shape().num_elements()); in Compute()708 const Tensor& tensor_stride = context->input(4); in Compute() local[all …]
334 const Tensor& tensor_stride = context->input(2);335 auto value_stride = tensor_stride.flat<int32>();336 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,