Home
last modified time | relevance | path

Searched refs:MatchingArraySize (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/neon/
Dtypes.h49 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
56 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
59 return MatchingArraySize(array1, index1, args...); in MatchingArraySize()
Ddepthwiseconv_float.h564 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3);
565 const int output_depth = MatchingArraySize(filter_dims, 0, output_dims, 0);
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h396 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
403 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
406 return MatchingArraySize(array1, index1, args...); in MatchingArraySize()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h412 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
419 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
422 return MatchingArraySize(array1, index1, args...); in MatchingArraySize()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h687 /* batches = */ MatchingArraySize(*output_dims[i], 3, input_dims, 3); in TensorFlowSplit()
688 /* height = */ MatchingArraySize(*output_dims[i], 2, input_dims, 2); in TensorFlowSplit()
689 /* width = */ MatchingArraySize(*output_dims[i], 1, input_dims, 1); in TensorFlowSplit()