Home
last modified time | relevance | path

Searched defs:output_strides (Results 1 – 11 of 11) sorted by relevance

/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h235 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestQS8() local
373 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestF16() local
524 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestF32() local
Dconstant-pad-operator-tester.h131 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestX32() local
/external/tensorflow/tensorflow/core/kernels/
Dreshape_util.cc55 gtl::InlinedVector<int64, 8> output_strides(output_rank); in operator ()() local
Dedit_distance_op.cc162 std::vector<int64> output_strides(output_shape.dims()); in Compute() local
Dsparse_reduce_op.cc195 gtl::InlinedVector<int64, 8> output_strides(reduction.group_by_dims.size()); in Compute() local
Dset_kernels.cc260 const auto output_strides = Strides(output_shape); in Compute() local
/external/libaom/libaom/av1/encoder/
Dcnn.h161 const int *output_strides; member
Dcnn.c1120 const int output_strides[1] = { out_stride }; in av1_cnn_predict_img() local
1138 const int output_strides[1] = { out_stride }; in av1_cnn_predict_img_highbd() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_slice_op.cc439 auto output_strides = CalculateTFStrides(size_dims); in ComputeMklSlice() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1188 const absl::Span<const int64> output_strides, in CopyDataToOutput()
1245 const absl::Span<const int64> output_strides, in CopyDataToOutput()
1413 const auto output_strides = ComputeStrides(output_lengths, output_literal); in HandleFft() local
/external/libaom/libaom/test/
Dcnn_test.cc54 const int output_strides[1] = { out_stride }; in RunCNNTest() local