Searched defs:input1_dims (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 572 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv()
585 inline void Div(const T* input1_data, const Dims<4>& input1_dims, in Div()
888 inline void Comparison(const T* input1_data, const Dims<4>& input1_dims, in Comparison()
900 const Dims<4>& input1_dims, int32 input1_offset, in Comparison()
924 const Dims<4>& input1_dims, in BroadcastComparison()
938 const Dims<4>& input1_dims, int32 input1_offset, in BroadcastComparison()
1166 const Dims<4>& input1_dims, int32 input1_offset, in Add()
1204 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add()
1220 const Dims<4>& input1_dims, int32 input1_offset, in BroadcastAdd()
1260 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n.cc | 47 TfLiteIntArray* input1_dims = input1->dims; in Prepare() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 873 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv()
930 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add()
946 const Dims<4>& input1_dims, int32 input1_offset, in Add()
984 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add()
999 void BroadcastAdd(const T* input1_data, const Dims<4>& input1_dims, in BroadcastAdd()
1013 const Dims<4>& input1_dims, int32 input1_offset, in BroadcastAdd()
1055 const uint8* input1_data, const Dims<4>& input1_dims, int32 input1_offset, in BroadcastAddFivefold()
1099 void BroadcastAdd(const T* input1_data, const Dims<4>& input1_dims, in BroadcastAdd()
1111 inline void Add(const int16* input1_data, const Dims<4>& input1_dims, in Add()
1138 inline void Sub(const float* input1_data, const Dims<4>& input1_dims, in Sub()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | common.h | 195 const Dims<N>& input1_dims, in NdArrayDescsForElementwiseBroadcast()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | common.h | 476 const Dims<N>& input1_dims, in NdArrayDescsForElementwiseBroadcast()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3556 void BroadcastDiv(const T* input1_data, const Dims<4>& input1_dims, in BroadcastDiv()
|