Home
last modified time | relevance | path

Searched defs:tensor_stride (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmaxpooling_op.cc298 const Tensor& tensor_stride = context->input(4); in Compute() local
416 const Tensor& tensor_stride = context->input(4); in Compute() local
522 const Tensor& tensor_stride = context->input(4); in Compute() local
740 const Tensor& tensor_stride = context->input(4); in Compute() local
904 const Tensor& tensor_stride = context->input(2); in Compute() local
1374 const Tensor& tensor_stride = context->input(2); in Compute() local
/aosp_15_r20/external/pytorch/torch/multiprocessing/
H A Dreductions.py126 tensor_stride, argument
158 tensor_stride, argument
/aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/
H A Dexecution_trace_observer.cpp360 std::string tensor_shape, tensor_stride, tensor_type, tensor_value; in convertIValue() local
452 std::string tensor_stride = "[]"; in convertIValue() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dlstm_utils.cc74 int tensor_stride, const Array& sub_array, in CopyArrayToSubArray()
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.cpp595 auto tensor_stride = invert_perm(op.stride_bytes); in allocate_or_resize_outputs() local