Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dmul.h43 const RuntimeShape& input1_shape, const int8_t* input1_data, in Mul()
57 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
86 const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
Dadd.h61 const RuntimeShape& input1_shape, const int8_t* input1_data, in Add()
78 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
/external/tensorflow/tensorflow/core/ops/
Dset_ops.cc58 ShapeHandle input1_shape = c->input(1); in __anon9b840f3f0102() local
76 ShapeHandle input1_shape = c->input(1); in __anon9b840f3f0102() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h526 const RuntimeShape& input1_shape, const T* input1_data, in Add()
539 const RuntimeShape& input1_shape, const float* input1_data, in Add()
634 const RuntimeShape& input1_shape, const uint8* input1_data, in Add()
650 const RuntimeShape& input1_shape, const int16* input1_data, in Add()
694 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
735 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
777 const RuntimeShape& input1_shape, in BroadcastAdd4DSlow()
935 const RuntimeShape& input1_shape, const T* input1_data, in Mul()
1030 const RuntimeShape& input1_shape, const uint8* input1_data, in Mul()
1095 const RuntimeShape& input1_shape, in BroadcastMul4DSlow()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Dlogical_test.cc28 LogicalOpModel(std::initializer_list<int> input1_shape, in LogicalOpModel()
Dselect_test.cc28 SelectOpModel(std::initializer_list<int> input1_shape, in SelectOpModel()
Dcomparisons_test.cc28 ComparisonOpModel(std::initializer_list<int> input1_shape, in ComparisonOpModel()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_binary.cc99 Shape input1_shape = input1_array.shape(); in EvaluateBinaryOperatorOnConstantInputs() local
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dcommon.h230 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape, in NdArrayDescsForElementwiseBroadcast()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h2624 const RuntimeShape& input1_shape, const float* input1_data, in Add()
2872 const RuntimeShape& input1_shape, const uint8* input1_data, in Add()
2889 const RuntimeShape& input1_shape, const int16* input1_data, in Add()
2926 const RuntimeShape& input1_shape, const int32* input1_data, in Add()
3046 const RuntimeShape& input1_shape, const float* input1_data, in Mul()
3106 const RuntimeShape& input1_shape, const int32* input1_data, in Mul()
3123 const RuntimeShape& input1_shape, in MulNoActivation()
3149 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
3170 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
3336 const RuntimeShape& input1_shape, const uint8* input1_data, in Mul()
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h2841 const RuntimeShape& input1_shape, const float* input1_data, in Add()
2985 const RuntimeShape& input1_shape, const uint8* input1_data, in Add()
3002 const RuntimeShape& input1_shape, const int16* input1_data, in Add()
3039 const RuntimeShape& input1_shape, const int32* input1_data, in Add()
3122 const RuntimeShape& input1_shape, const float* input1_data, in Mul()
3182 const RuntimeShape& input1_shape, const int32* input1_data, in Mul()
3199 const RuntimeShape& input1_shape, in MulNoActivation()
3225 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
3246 const RuntimeShape& input1_shape, const int16* input1_data, in Mul()
3412 const RuntimeShape& input1_shape, const uint8* input1_data, in Mul()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h511 const RuntimeShape& input0_shape, const RuntimeShape& input1_shape, in NdArrayDescsForElementwiseBroadcast()