Home
last modified time | relevance | path

Searched defs:input_strides (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dreshape_util.cc47 gtl::InlinedVector<int64, 8> input_strides(input_rank); in operator ()() local
Dset_kernels.cc204 const VarDimArray& input_strides, in PopulateFromDenseGroup()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_transpose.cc59 int input_strides[4]; in Transpose() local
/external/XNNPACK/test/
Dconstant-pad-operator-tester.h130 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX32() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_slice_op.cc400 memory::dims input_dims, input_strides; in ComputeMklSlice() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1135 const absl::Span<const int64> input_strides, in CopyDataFromInput()
1225 const absl::Span<const int64> input_strides, in CopyDataFromInput()
1412 const auto input_strides = ComputeStrides(input_lengths, input_literal); in HandleFft() local
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1207 mkldnn_dim_t* input_strides = new mkldnn_dim_t[kNumDims]; in CreateBlockedMemDescHelper() local
/external/tensorflow/tensorflow/lite/toco/
Dtooling_util.cc2151 int input_strides[4]; in ShuffleArrayTemplate() local