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.h562 const int batches = MatchingArraySize(input_dims, 3, output_dims, 3);
563 const int output_depth = MatchingArraySize(filter_dims, 0, output_dims, 0);
/external/tensorflow/tensorflow/lite/micro/kernels/ceva/
Dtypes.h605 int MatchingArraySize(const ArrayType1& array1, int index1,
612 int MatchingArraySize(const ArrayType1& array1, int index1,
615 return MatchingArraySize(array1, index1, args...);
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h456 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
463 int MatchingArraySize(const ArrayType1& array1, int index1, in MatchingArraySize() function
466 return MatchingArraySize(array1, index1, args...); in MatchingArraySize()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h737 /* batches = */ MatchingArraySize(*output_dims[i], 3, input_dims, 3); in TensorFlowSplit()
738 /* height = */ MatchingArraySize(*output_dims[i], 2, input_dims, 2); in TensorFlowSplit()
739 /* width = */ MatchingArraySize(*output_dims[i], 1, input_dims, 1); in TensorFlowSplit()