| /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() argument
579 BroadcastDiv4DSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastDiv()
585 inline void Div(const T* input1_data, const Dims<4>& input1_dims, in Div() argument
592 Div(op_params, DimsToShape(input1_dims), input1_data, in Div()
888 inline void Comparison(const T* input1_data, const Dims<4>& input1_dims, in Comparison() argument
893 ComparisonImpl<T, F>(op_params, DimsToShape(input1_dims), input1_data, in Comparison()
900 const Dims<4>& input1_dims, int32 input1_offset, in Comparison() argument
917 ComparisonWithScaling<T, F>(op_params, DimsToShape(input1_dims), input1_data, in Comparison()
924 const Dims<4>& input1_dims, in BroadcastComparison() argument
930 BroadcastComparison4DSlowImpl<T, F>(op_params, DimsToShape(input1_dims), in BroadcastComparison()
[all …]
|
| /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() argument
880 BroadcastDiv4DSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastDiv()
930 void Add(const float* input1_data, const Dims<4>& input1_dims, in Add() argument
939 Add(op_params, DimsToShape(input1_dims), input1_data, in Add()
946 const Dims<4>& input1_dims, int32 input1_offset, in Add() argument
978 Add(op_params, DimsToShape(input1_dims), input1_data, in Add()
984 void Add(const int32* input1_data, const Dims<4>& input1_dims, in Add() argument
993 Add(op_params, DimsToShape(input1_dims), input1_data, in Add()
999 void BroadcastAdd(const T* input1_data, const Dims<4>& input1_dims, in BroadcastAdd() argument
1006 BroadcastAdd4DSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastAdd()
[all …]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | common.h | 195 const Dims<N>& input1_dims, in NdArrayDescsForElementwiseBroadcast() argument
205 desc1_out->extents[i] = input1_dims.sizes[i]; in NdArrayDescsForElementwiseBroadcast()
206 desc1_out->strides[i] = input1_dims.strides[i]; in NdArrayDescsForElementwiseBroadcast()
214 const int extent1 = ArraySize(input1_dims, i); in NdArrayDescsForElementwiseBroadcast()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | add_n.cc | 47 TfLiteIntArray* input1_dims = input1->dims; in Prepare() local
48 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input1_dims); in Prepare()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | common.h | 476 const Dims<N>& input1_dims, in NdArrayDescsForElementwiseBroadcast() argument
486 desc1_out->extents[i] = input1_dims.sizes[i]; in NdArrayDescsForElementwiseBroadcast()
487 desc1_out->strides[i] = input1_dims.strides[i]; in NdArrayDescsForElementwiseBroadcast()
495 const int extent1 = ArraySize(input1_dims, i); 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() argument
3563 BroadcastDiv4DSlow(op_params, DimsToShape(input1_dims), input1_data, in BroadcastDiv()
|