Home
last modified time | relevance | path

Searched defs:input1_shape (Results 1 – 25 of 29) 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 …]
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()
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()
Dadd.h29 const RuntimeShape& input1_shape, const T* input1_data, in Add()
116 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Add()
132 const RuntimeShape& input1_shape, in AddGeneralParamScale()
199 const RuntimeShape& input1_shape, const T* input1_data, in BroadcastAdd4DSlow()
244 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.h75 const RuntimeShape& input1_shape, const uint8_t* input1_data, in Div()
87 const RuntimeShape& input1_shape, const int8_t* input1_data, in Div()
220 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()
Dbinary_function.h64 inline void BinaryFunction(const RuntimeShape& input1_shape, in BinaryFunction()
Dreference_ops.h223 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
242 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
271 const RuntimeShape& input1_shape, const int16_t* input1_data, in Sub16()
1113 void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum()
1127 inline void Minimum(const RuntimeShape& input1_shape, const T* input1_data, in Minimum()
1135 void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum()
1149 inline void Maximum(const RuntimeShape& input1_shape, const T* input1_data, in Maximum()
1157 void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax()
1167 inline void ArgMax(const RuntimeShape& input1_shape, const T1* input1_data, in ArgMax()
1347 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()
/external/XNNPACK/src/operators/
Dbinary-elementwise-nd.c771 const size_t* input1_shape, in setup_binary_elementwise_nd()
961 const size_t* input1_shape, in setup_binary_elementwise_nd_f16()
992 const size_t* input1_shape, in setup_binary_elementwise_nd_f32()
1017 const size_t* input1_shape, in xnn_setup_add_nd_qs8()
1041 const size_t* input1_shape, in xnn_setup_add_nd_qu8()
1065 const size_t* input1_shape, in xnn_setup_add_nd_f16()
1085 const size_t* input1_shape, in xnn_setup_add_nd_f32()
1105 const size_t* input1_shape, in xnn_setup_divide_nd_f32()
1125 const size_t* input1_shape, in xnn_setup_maximum_nd_f32()
1145 const size_t* input1_shape, in xnn_setup_minimum_nd_f32()
[all …]
/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 __anon19a2ed6b0102() local
76 ShapeHandle input1_shape = c->input(1); in __anon19a2ed6b0102() local
/external/tensorflow/tensorflow/lite/kernels/gradient/
Dbcast_grad_args.cc45 const RuntimeShape input1_shape = GetTensorShape(input1); in Prepare() local
71 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/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/XNNPACK/test/
Dbinary-elementwise-operator-tester.h40 inline BinaryElementwiseOperatorTester& input1_shape(std::initializer_list<size_t> input1_shape) { in input1_shape() function
46 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
/external/ComputeLibrary/src/runtime/CL/functions/
DCLWinogradConvolutionLayer.cpp202 …const TensorShape input1_shape = misc::shape_calculator::compute_winograd_filter_transform_shape(*… in validate() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1847 const RuntimeShape& input1_shape, const float* input1_data, in Add()
2084 const RuntimeShape& input1_shape, const uint8* input1_data, in Add()
2101 const RuntimeShape& input1_shape, const int16* input1_data, in Add()
2139 const ArithmeticParams& params, const RuntimeShape& input1_shape, in Add()
2173 const ArithmeticParams& params, const RuntimeShape& input1_shape, in BroadcastAddDispatch()
2272 const RuntimeShape& input1_shape, const float* input1_data, in Mul()
2283 const RuntimeShape& input1_shape, const int32* input1_data, in Mul()
2300 const RuntimeShape& input1_shape, in MulNoActivation()
2327 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
2348 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
[all …]

12