Searched defs:input2_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | legacy_reference_ops.h | 573 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
586 const T* input2_data, const Dims<4>& input2_dims, in Div()
889 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
902 const T* input2_data, const Dims<4>& input2_dims, in Comparison()
926 const Dims<4>& input2_dims, bool* output_data, in BroadcastComparison()
941 const Dims<4>& input2_dims, int32 input2_offset, in BroadcastComparison()
1168 const uint8* input2_data, const Dims<4>& input2_dims, in Add()
1205 const int32* input2_data, const Dims<4>& input2_dims, in Add()
1222 const uint8* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
1261 const float* input2_data, const Dims<4>& input2_dims, in Add()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 874 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
931 const float* input2_data, const Dims<4>& input2_dims, in Add()
948 const uint8* input2_data, const Dims<4>& input2_dims, in Add()
985 const int32* input2_data, const Dims<4>& input2_dims, in Add()
1000 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
1015 const uint8* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
1057 const Dims<4>& input2_dims, int32 input2_offset, int32 input2_multiplier, in BroadcastAddFivefold()
1100 const T* input2_data, const Dims<4>& input2_dims, in BroadcastAdd()
1113 const Dims<4>& input2_dims, int input2_shift, in Add()
1139 const float* input2_data, const Dims<4>& input2_dims, in Sub()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 3557 const T* input2_data, const Dims<4>& input2_dims, in BroadcastDiv()
|