Home
last modified time | relevance | path

Searched defs:input1_shape (Results 1 – 25 of 32) 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()
150 const RuntimeShape& input1_shape, in BroadcastSubSlow()
189 const RuntimeShape& input1_shape, const T* input1_data, in BroadcastSubSlow()
226 const RuntimeShape& input1_shape, in BroadcastSub16POTSlow()
270 const RuntimeShape& input1_shape, in BroadcastQuantSubSlow()
354 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Sub()
370 const RuntimeShape& input1_shape, const int8_t* input1_data, in Sub()
[all …]
Dmul.h49 const RuntimeShape& input1_shape, const T* input1_data, in Mul()
66 const RuntimeShape& input1_shape, in Mul()
80 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Mul()
92 const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
Darg_min_max.h36 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax()
78 void ArgMinMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMinMax()
Dadd.h30 const RuntimeShape& input1_shape, const T* input1_data, in Add()
117 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Add()
133 const RuntimeShape& input1_shape, in AddGeneralParamScale()
200 const RuntimeShape& input1_shape, const T* input1_data, in BroadcastAdd4DSlow()
245 const RuntimeShape& input1_shape, const T* input1_data, in BroadcastAdd4DSlow()
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.h79 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Div()
91 const RuntimeShape& input1_shape, const int8_t* input1_data, in Div()
228 const RuntimeShape& input1_shape, const T* input1_data, in Div()
Dbroadcast_args.h25 void BroadcastArgs(const RuntimeShape& input1_shape, const T* input1_data, in BroadcastArgs()
Dreference_ops.h239 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
258 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
287 const RuntimeShape& input1_shape, const int16_t* input1_data, in Sub16()
708 void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum()
722 inline void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum()
730 void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum()
744 inline void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum()
752 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax()
762 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax()
843 inline void Pow(const RuntimeShape& input1_shape, const T* input1_data, in Pow()
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()
Dbinary_function.h75 inline void BinaryFunction(const RuntimeShape& input1_shape, in BinaryFunction()
/external/XNNPACK/src/operators/
Dbinary-elementwise-nd.c843 const size_t* input1_shape, in setup_binary_elementwise_nd()
1053 const size_t* input1_shape, in setup_binary_elementwise_nd_f16()
1083 const size_t* input1_shape, in setup_binary_elementwise_nd_f32()
1107 const size_t* input1_shape, in xnn_setup_add_nd_f16()
1127 const size_t* input1_shape, in xnn_setup_add_nd_f32()
1147 const size_t* input1_shape, in xnn_setup_add_nd_qs8()
1170 const size_t* input1_shape, in xnn_setup_add_nd_qu8()
1193 const size_t* input1_shape, in xnn_setup_divide_nd_f16()
1213 const size_t* input1_shape, in xnn_setup_divide_nd_f32()
1233 const size_t* input1_shape, in xnn_setup_maximum_nd_f16()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h48 const RuntimeShape& input1_shape, const T* input1_data, in Mul()
62 const RuntimeShape& input1_shape, const int16_t* input1_data, in Mul()
92 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
Dadd.h50 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastBinaryFunction4DSlow()
119 const RuntimeShape& input1_shape, const int8_t* input1_data, in Add()
131 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
/external/tensorflow/tensorflow/core/ops/
Dset_ops.cc58 ShapeHandle input1_shape = c->input(1); in __anonbfa34e750102() local
76 ShapeHandle input1_shape = c->input(1); in __anonbfa34e750102() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Dsub.h125 const RuntimeShape& input1_shape, in BroadcastSubFiveFold()
194 const RuntimeShape& input1_shape, const int16* input1_data, in Sub()
211 const RuntimeShape& input1_shape, in BroadcastSubDispatch()
Dmul.h233 const RuntimeShape& input1_shape, const int8* input1_data, in Mul()
246 const RuntimeShape& input1_shape, in BroadcastMulDispatch()
Dadd.h459 const RuntimeShape& input1_shape, const int8* input1_data, in Add()
476 const RuntimeShape& input1_shape, const int16* input1_data, in Add()
493 const RuntimeShape& input1_shape, in BroadcastAddDispatch()
/external/tensorflow/tensorflow/lite/kernels/gradient/
Dbcast_grad_args.cc46 const RuntimeShape input1_shape = GetTensorShape(input1); in Prepare() local
72 const RuntimeShape input1_shape = GetTensorShape(input1); in Invoke() local
/external/tensorflow/tensorflow/lite/kernels/
Dselect_test.cc33 BaseSelectOpModel(std::initializer_list<int> input1_shape, in BaseSelectOpModel()
62 SelectOpModel(std::initializer_list<int> input1_shape, in SelectOpModel()
75 SelectV2OpModel(std::initializer_list<int> input1_shape, in SelectV2OpModel()
Dlogical_test.cc31 LogicalOpModel(std::initializer_list<int> input1_shape, in LogicalOpModel()
/external/XNNPACK/test/
Dsubgraph-binary-tester.h43 std::vector<size_t> input1_shape = RandomShape(); in SetUp() local
Dbinary-elementwise-operator-tester.h39 inline BinaryElementwiseOperatorTester& input1_shape(std::initializer_list<size_t> input1_shape) { in input1_shape() function
45 inline const std::vector<size_t>& input1_shape() const { in input1_shape() function
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_binary.cc100 Shape input1_shape = input1_array.shape(); in EvaluateBinaryOperatorOnConstantInputs() local

12