Home
last modified time | relevance | path

Searched defs:input1_shape (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsub.h33 const RuntimeShape& input1_shape, in SubNonBroadcast()
49 const RuntimeShape& input1_shape, in SubNonBroadcast()
70 const RuntimeShape& input1_shape, in BroadcastSubSlow()
110 const RuntimeShape& input1_shape, in BroadcastSubSlow()
167 const RuntimeShape& input1_shape, in BroadcastSubSlow()
207 const RuntimeShape& input1_shape, in BroadcastSubSlow()
261 const RuntimeShape& input1_shape, in BroadcastSubSlow()
300 const RuntimeShape& input1_shape, const T* input1_data, in BroadcastSubSlow()
337 const RuntimeShape& input1_shape, in BroadcastSub16POTSlow()
447 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Sub()
[all …]
Dadd.h27 const RuntimeShape& input1_shape, const T* input1_data, in Add()
40 const RuntimeShape& input1_shape, const float* input1_data, in Add()
125 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Add()
141 const RuntimeShape& input1_shape, in AddGeneralParamScale()
206 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
246 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
291 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
Dmul.h46 const RuntimeShape& input1_shape, const T* input1_data, in Mul()
63 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Mul()
75 const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
Dcomparisons.h58 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonImpl()
70 const RuntimeShape& input1_shape, in Comparison()
81 const ComparisonParams& op_params, const RuntimeShape& input1_shape, in ComparisonWithScaling()
156 const RuntimeShape& input1_shape, in BroadcastComparison4DSlow()
Ddiv.h62 const RuntimeShape& input1_shape, const uint8* input1_data, in Div()
175 const RuntimeShape& input1_shape, const T* input1_data, in Div()
Darg_min_max.h25 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax()
Dbinary_function.h64 inline void BinaryFunction(const RuntimeShape& input1_shape, in BinaryFunction()
Dbatch_to_space_nd.h53 const RuntimeShape input1_shape = in BatchToSpaceND() local
Dspace_to_batch_nd.h56 const RuntimeShape input1_shape = in SpaceToBatchND() local
Dstring_comparisons.h41 const RuntimeShape& input1_shape, in ComparisonStringImpl()
Dreference_ops.h221 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
240 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
269 const RuntimeShape& input1_shape, const int16_t* input1_data, in Sub16()
1422 void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum()
1436 inline void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum()
1444 void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum()
1458 inline void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum()
1466 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax()
1476 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax()
1648 inline void Pow(const RuntimeShape& input1_shape, const T* input1_data, in Pow()
/external/XNNPACK/src/operators/
Dbinary-elementwise-nd.c376 const size_t* input1_shape, in setup_binary_elementwise_nd()
573 const size_t* input1_shape, in setup_binary_elementwise_nd_f16()
604 const size_t* input1_shape, in setup_binary_elementwise_nd_f32()
629 const size_t* input1_shape, in xnn_setup_add_nd_qs8()
653 const size_t* input1_shape, in xnn_setup_add_nd_f16()
673 const size_t* input1_shape, in xnn_setup_add_nd_f32()
693 const size_t* input1_shape, in xnn_setup_divide_nd_f32()
713 const size_t* input1_shape, in xnn_setup_maximum_nd_f32()
733 const size_t* input1_shape, in xnn_setup_minimum_nd_f32()
753 const size_t* input1_shape, in xnn_setup_multiply_nd_f16()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Dcomparisons.cc48 RuntimeShape input1_shape = tflite::micro::GetTensorShape(input1); in EqualEval() local
153 RuntimeShape input1_shape = tflite::micro::GetTensorShape(input1); in NotEqualEval() local
257 RuntimeShape input1_shape = tflite::micro::GetTensorShape(input1); in GreaterEval() local
348 RuntimeShape input1_shape = tflite::micro::GetTensorShape(input1); in GreaterEqualEval() local
439 RuntimeShape input1_shape = tflite::micro::GetTensorShape(input1); in LessEval() local
530 RuntimeShape input1_shape = tflite::micro::GetTensorShape(input1); in LessEqualEval() local
Dlogical_test.cc82 const int input1_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
105 const int input1_shape[] = {4, 1, 1, 1, 4}; in TF_LITE_MICRO_TEST() local
Darg_min_max.cc35 inline void ArgMinMaxHelper(const RuntimeShape& input1_shape, in ArgMinMaxHelper()
Dconcatenation_test.cc191 const int input1_shape[] = {3, 2, 1, 2}; in TF_LITE_MICRO_TEST() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h46 const RuntimeShape& input1_shape, const T* input1_data, in Mul()
60 const RuntimeShape& input1_shape, const int16_t* input1_data, in Mul()
90 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
Dadd.h49 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastBinaryFunction4DSlow()
118 const RuntimeShape& input1_shape, const int8_t* input1_data, in Add()
130 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
/external/tensorflow/tensorflow/core/ops/
Dset_ops.cc58 ShapeHandle input1_shape = c->input(1); in __anonda6063e60102() local
76 ShapeHandle input1_shape = c->input(1); in __anonda6063e60102() local
/external/tensorflow/tensorflow/lite/kernels/
Dselect_test.cc32 BaseSelectOpModel(std::initializer_list<int> input1_shape, in BaseSelectOpModel()
61 SelectOpModel(std::initializer_list<int> input1_shape, in SelectOpModel()
74 SelectV2OpModel(std::initializer_list<int> input1_shape, in SelectV2OpModel()
Dlogical_test.cc31 LogicalOpModel(std::initializer_list<int> input1_shape, in LogicalOpModel()
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h40 inline BinaryElementwiseOperatorTester& input1_shape(std::initializer_list<size_t> input1_shape) { in input1_shape() argument
46 inline const std::vector<size_t>& input1_shape() const { in input1_shape() function
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dmul.h233 const RuntimeShape& input1_shape, const int8* input1_data, in Mul()
246 const RuntimeShape& input1_shape, in BroadcastMulDispatch()
Dadd.h268 const RuntimeShape& input1_shape, const int8* input1_data, in Add()
285 const RuntimeShape& input1_shape, in BroadcastAddDispatch()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_binary.cc100 Shape input1_shape = input1_array.shape(); in EvaluateBinaryOperatorOnConstantInputs() local

12