Home
last modified time | relevance | path

Searched refs:MatchingFlatSizeSkipDim (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h541 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
552 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
560 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1); in MatchingFlatSizeSkipDim()
564 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
573 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2); in MatchingFlatSizeSkipDim()
577 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
587 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2, in MatchingFlatSizeSkipDim()
606 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim() function
617 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim() function
626 return MatchingFlatSizeSkipDim(shape, skip_dim, check_shape_1); in MatchingFlatSizeSkipDim()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h557 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
568 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
576 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1); in MatchingFlatSizeSkipDim()
580 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
589 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2); in MatchingFlatSizeSkipDim()
593 inline int MatchingFlatSizeSkipDim(const Dims<N>& dims, int skip_dim, in MatchingFlatSizeSkipDim() function
603 return MatchingFlatSizeSkipDim(dims, skip_dim, check_dims_1, check_dims_2, in MatchingFlatSizeSkipDim()
622 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim() function
633 inline int MatchingFlatSizeSkipDim(const RuntimeShape& shape, int skip_dim, in MatchingFlatSizeSkipDim() function
642 return MatchingFlatSizeSkipDim(shape, skip_dim, check_shape_1); in MatchingFlatSizeSkipDim()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsoftmax.h33 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
80 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
Dreference_ops.h476 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
501 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
2086 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell()
2547 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LocalResponseNormalization()
2572 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
2622 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
4324 MatchingFlatSizeSkipDim(input_x_shape, 0, input_y_shape, output_shape); in RankOneSelect()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dsoftmax.h44 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
/external/tensorflow/tensorflow/lite/kernels/
Dl2norm.cc114 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Eval()
Dactivations.cc868 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmaxEval()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2568 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
2596 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
3724 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell()
4472 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
4670 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
4723 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h2736 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
2813 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in L2Normalization()
3840 const int outer_size = MatchingFlatSizeSkipDim( in LstmCell()
4597 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in Softmax()
4808 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()
4984 MatchingFlatSizeSkipDim(input_shape, trailing_dim, output_shape); in LogSoftmax()