/external/tensorflow/tensorflow/contrib/lite/kernels/ |
resize_bilinear_test.cc | 37 output_ = AddOutput(TensorType_FLOAT32); // Always float. 62 ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 1, 2, 1}}); 68 ResizeBilinearOpModel const_m({TensorType_FLOAT32, {1, 1, 2, 1}}, {1, 3}); 75 ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 2, 1, 1}}); 81 ResizeBilinearOpModel const_m({TensorType_FLOAT32, {1, 2, 1, 1}}, {3, 1}); 88 ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}); 101 ResizeBilinearOpModel const_m({TensorType_FLOAT32, {1, 2, 2, 1}}, {3, 3}); 115 ResizeBilinearOpModel m({TensorType_FLOAT32, {2, 2, 2, 1}}); 133 ResizeBilinearOpModel const_m({TensorType_FLOAT32, {2, 2, 2, 1}}, {3, 3}); 152 ResizeBilinearOpModel m({TensorType_FLOAT32, {1, 2, 2, 2}}) [all...] |
squeeze_test.cc | 63 FloatSqueezeOpModel m({TensorType_FLOAT32, {1, 24, 1}}, 64 {TensorType_FLOAT32, {24}}, {}); 79 FloatSqueezeOpModel m({TensorType_FLOAT32, {1, 24, 1}}, 80 {TensorType_FLOAT32, {24}}, {2}); 95 FloatSqueezeOpModel m({TensorType_FLOAT32, {1, 24, 1}}, 96 {TensorType_FLOAT32, {24}}, {-1, 0}); 109 FloatSqueezeOpModel m({TensorType_FLOAT32, {1, 1, 1, 1, 1, 1, 1}}, 110 {TensorType_FLOAT32, {1}}, {});
|
optional_tensor_test.cc | 41 input_ = AddInput(TensorType_FLOAT32); 46 input_to_input_weights_ = AddInput(TensorType_FLOAT32); 49 input_to_forget_weights_ = AddInput(TensorType_FLOAT32); 50 input_to_cell_weights_ = AddInput(TensorType_FLOAT32); 51 input_to_output_weights_ = AddInput(TensorType_FLOAT32); 56 recurrent_to_input_weights_ = AddInput(TensorType_FLOAT32); 59 recurrent_to_forget_weights_ = AddInput(TensorType_FLOAT32); 60 recurrent_to_cell_weights_ = AddInput(TensorType_FLOAT32); 61 recurrent_to_output_weights_ = AddInput(TensorType_FLOAT32); 67 cell_to_input_weights_ = AddInput(TensorType_FLOAT32); [all...] |
conv_test.cc | 54 if (input.type == TensorType_FLOAT32) { 55 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); 66 if (input.type != TensorType_FLOAT32) { 120 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 1}}, 121 {TensorType_FLOAT32, {3, 2, 2, 1}}, 122 {TensorType_FLOAT32, {}}); 150 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {1, 3, 6, 1}}, 151 {TensorType_FLOAT32, {1, 2, 2, 1}}, 152 {TensorType_FLOAT32, {}}, 183 {TensorType_FLOAT32, [all...] |
mul_test.cc | 73 FloatMulOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, 74 {TensorType_FLOAT32, {1, 2, 2, 1}}, 75 {TensorType_FLOAT32, {}}, ActivationFunctionType_NONE); 85 {TensorType_FLOAT32, {1, 2, 2, 1}}, {TensorType_FLOAT32, {1, 2, 2, 1}}, 86 {TensorType_FLOAT32, {}}, ActivationFunctionType_RELU_N1_TO_1); 98 FloatMulOpModel m({TensorType_FLOAT32, test_shapes[i]}, 99 {TensorType_FLOAT32, test_shapes[i]}, 100 {TensorType_FLOAT32, {}}, ActivationFunctionType_NONE); 115 FloatMulOpModel m({TensorType_FLOAT32, test_shapes[i]} [all...] |
mean_test.cc | 81 MeanOpConstModel m({TensorType_FLOAT32, {4, 3, 2}}, {TensorType_FLOAT32, {2}}, 93 MeanOpConstModel m({TensorType_FLOAT32, {4, 3, 2}}, {TensorType_FLOAT32, {3}}, 106 MeanOpDynamicModel m({TensorType_FLOAT32, {4, 3, 2}}, 107 {TensorType_FLOAT32, {2}}, {TensorType_INT32, {4}}, 121 MeanOpDynamicModel m({TensorType_FLOAT32, {4, 3, 2}}, 122 {TensorType_FLOAT32, {3}}, {TensorType_INT32, {2}},
|
basic_rnn_test.cc | 129 input_ = AddInput(TensorType_FLOAT32); 130 weights_ = AddInput(TensorType_FLOAT32); 131 recurrent_weights_ = AddInput(TensorType_FLOAT32); 132 bias_ = AddInput(TensorType_FLOAT32); 133 hidden_state_ = AddOutput(TensorType_FLOAT32); 134 output_ = AddOutput(TensorType_FLOAT32);
|
depthwise_conv_test.cc | 38 if (input.type == TensorType_FLOAT32) { 39 bias_ = AddInput({TensorType_FLOAT32, {bias_size}}); 50 if (input.type != TensorType_FLOAT32) { 94 DepthwiseConvolutionOpModel m({TensorType_FLOAT32, {1, 3, 2, 2}}, 95 {TensorType_FLOAT32, {1, 2, 2, 4}}, 96 {TensorType_FLOAT32, {}});
|
pooling_test.cc | 79 /*input=*/{TensorType_FLOAT32, {1, 2, 4, 1}}, 81 /*output=*/{TensorType_FLOAT32, {}}); 111 /*input=*/{TensorType_FLOAT32, {1, 2, 4, 1}}, 113 /*output=*/{TensorType_FLOAT32, {}}); 143 /*input=*/{TensorType_FLOAT32, {1, 2, 4, 1}}, 145 /*output=*/{TensorType_FLOAT32, {}});
|
l2norm_test.cc | 30 input_ = AddInput(TensorType_FLOAT32); 31 output_ = AddOutput(TensorType_FLOAT32);
|
lstm_test.cc | 43 input_ = AddInput(TensorType_FLOAT32); 48 input_to_input_weights_ = AddInput(TensorType_FLOAT32); 51 input_to_forget_weights_ = AddInput(TensorType_FLOAT32); 52 input_to_cell_weights_ = AddInput(TensorType_FLOAT32); 53 input_to_output_weights_ = AddInput(TensorType_FLOAT32); 58 recurrent_to_input_weights_ = AddInput(TensorType_FLOAT32); 61 recurrent_to_forget_weights_ = AddInput(TensorType_FLOAT32); 62 recurrent_to_cell_weights_ = AddInput(TensorType_FLOAT32); 63 recurrent_to_output_weights_ = AddInput(TensorType_FLOAT32); 69 cell_to_input_weights_ = AddInput(TensorType_FLOAT32); [all...] |
unidirectional_sequence_lstm_test.cc | 46 input_ = AddInput(TensorType_FLOAT32); 51 input_to_input_weights_ = AddInput(TensorType_FLOAT32); 54 input_to_forget_weights_ = AddInput(TensorType_FLOAT32); 55 input_to_cell_weights_ = AddInput(TensorType_FLOAT32); 56 input_to_output_weights_ = AddInput(TensorType_FLOAT32); 61 recurrent_to_input_weights_ = AddInput(TensorType_FLOAT32); 64 recurrent_to_forget_weights_ = AddInput(TensorType_FLOAT32); 65 recurrent_to_cell_weights_ = AddInput(TensorType_FLOAT32); 66 recurrent_to_output_weights_ = AddInput(TensorType_FLOAT32); 72 cell_to_input_weights_ = AddInput(TensorType_FLOAT32); [all...] |
add_test.cc | 73 FloatAddOpModel m({TensorType_FLOAT32, {1, 2, 2, 1}}, 74 {TensorType_FLOAT32, {1, 2, 2, 1}}, 75 {TensorType_FLOAT32, {}}, ActivationFunctionType_NONE); 84 {TensorType_FLOAT32, {1, 2, 2, 1}}, {TensorType_FLOAT32, {1, 2, 2, 1}}, 85 {TensorType_FLOAT32, {}}, ActivationFunctionType_RELU_N1_TO_1); 96 FloatAddOpModel m({TensorType_FLOAT32, test_shapes[i]}, 97 {TensorType_FLOAT32, test_shapes[i]}, 98 {TensorType_FLOAT32, {}}, ActivationFunctionType_NONE); 112 FloatAddOpModel m({TensorType_FLOAT32, test_shapes[i]} [all...] |
unidirectional_sequence_rnn_test.cc | 133 input_ = AddInput(TensorType_FLOAT32); 134 weights_ = AddInput(TensorType_FLOAT32); 135 recurrent_weights_ = AddInput(TensorType_FLOAT32); 136 bias_ = AddInput(TensorType_FLOAT32); 137 hidden_state_ = AddOutput(TensorType_FLOAT32); 138 output_ = AddOutput(TensorType_FLOAT32);
|
activations_test.cc | 92 /*input=*/{TensorType_FLOAT32, {1, 2, 4, 1}}); 106 /*input=*/{TensorType_FLOAT32, {1, 2, 4, 1}}); 120 /*input=*/{TensorType_FLOAT32, {1, 2, 4, 1}}); 134 /*input=*/{TensorType_FLOAT32, {1, 2, 4, 1}}); 148 /*input=*/{TensorType_FLOAT32, {1, 2, 4, 1}}); 182 /*input=*/{TensorType_FLOAT32, {1, 2, 1, 4}}); 195 /*input=*/{TensorType_FLOAT32, {4, 1, 1, 2}}); 251 /*input=*/{TensorType_FLOAT32, {2, 4}}); 264 /*input=*/{TensorType_FLOAT32, {4, 2}});
|
batch_to_space_nd_test.cc | 62 input_ = AddInput(TensorType_FLOAT32); 65 output_ = AddOutput(TensorType_FLOAT32); 85 input_ = AddInput(TensorType_FLOAT32); 88 output_ = AddOutput(TensorType_FLOAT32);
|
pad_test.cc | 56 input_ = AddInput(TensorType_FLOAT32); 58 output_ = AddOutput(TensorType_FLOAT32); 77 input_ = AddInput(TensorType_FLOAT32); 79 output_ = AddOutput(TensorType_FLOAT32);
|
bidirectional_sequence_rnn_test.cc | [all...] |
concatenation_test.cc | 74 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2}}, /*axis=*/1, 82 ConcatenationOpModel m0({TensorType_FLOAT32, {1}}, /*axis=*/0, 90 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 3}}, /*axis=*/0, 102 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 3}}, /*axis=*/0, 110 ConcatenationOpModel m0_negative({TensorType_FLOAT32, {2, 3}}, /*axis=*/-2, 118 ConcatenationOpModel m1({TensorType_FLOAT32, {2, 3}}, /*axis=*/1, 126 ConcatenationOpModel m1_negative({TensorType_FLOAT32, {2, 3}}, /*axis=*/-1, 136 ConcatenationOpModel m0({TensorType_FLOAT32, {2, 1, 2}}, /*axis=*/2,
|
exp_test.cc | 54 ExpOpModel m({TensorType_FLOAT32, {3, 1, 2}}, TensorType_FLOAT32);
|
local_response_norm_test.cc | 30 input_ = AddInput(TensorType_FLOAT32); 31 output_ = AddOutput(TensorType_FLOAT32);
|
reshape_test.cc | 30 input_ = AddInput(TensorType_FLOAT32); 31 output_ = AddOutput(TensorType_FLOAT32);
|
gather_test.cc | 71 GatherOpModel m({2, 2}, TensorType_FLOAT32, {2}); 80 GatherOpModel m({2, 2}, TensorType_FLOAT32, {}); 91 GatherOpModel m({3}, TensorType_FLOAT32, {}); 100 GatherOpModel m({1, 2, 2}, TensorType_FLOAT32, {2}); 110 GatherOpModel m({4, 1}, TensorType_FLOAT32, {2});
|
svdf_test.cc | 137 input_ = AddInput(TensorType_FLOAT32); 138 weights_feature_ = AddInput(TensorType_FLOAT32); 139 weights_time_ = AddInput(TensorType_FLOAT32); 141 state_ = AddOutput(TensorType_FLOAT32); 142 output_ = AddOutput(TensorType_FLOAT32);
|
space_to_batch_nd_test.cc | 62 input_ = AddInput(TensorType_FLOAT32); 65 output_ = AddOutput(TensorType_FLOAT32); 85 input_ = AddInput(TensorType_FLOAT32); 88 output_ = AddOutput(TensorType_FLOAT32);
|