HomeSort by relevance Sort by last modified time
    Searched refs:TensorType_FLOAT32 (Results 1 - 25 of 44) sorted by null

1 2

  /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);

Completed in 371 milliseconds

1 2