Home
last modified time | relevance | path

Searched refs:kScalarShape (Results 1 – 25 of 32) sorted by relevance

12

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Darithmetic_builder.cc50 kScalarShape, reinterpret_cast<char*>(&output_min_), sizeof(output_min_)); in PopulateSubGraph()
52 kScalarShape, reinterpret_cast<char*>(&output_max_), sizeof(output_max_)); in PopulateSubGraph()
68 const auto& math_out_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
69 const auto& math_out_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
82 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
83 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
88 auto result_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
89 auto result_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
102 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
103 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dpack_builder.cc48 kScalarShape, reinterpret_cast<char*>(&axis), sizeof(axis)); in PopulateSubGraph()
69 kScalarShape, reinterpret_cast<char*>(&minima_[i]), sizeof(minima_[i])); in PopulateSubGraph()
76 kScalarShape, reinterpret_cast<char*>(&maxima_[i]), sizeof(maxima_[i])); in PopulateSubGraph()
95 kScalarShape, reinterpret_cast<char*>(&output_min), sizeof(output_min)); in PopulateSubGraph()
97 kScalarShape, reinterpret_cast<char*>(&output_max), sizeof(output_max)); in PopulateSubGraph()
99 const auto& pack_out_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
100 const auto& pack_out_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
114 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
115 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dpool_2d_builder.cc70 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
71 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
78 const auto& pool_out_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
79 const auto& pool_out_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
85 kScalarShape, (char*)&output_min_, sizeof(output_min_)); in PopulateSubGraph()
87 kScalarShape, (char*)&output_max_, sizeof(output_max_)); in PopulateSubGraph()
100 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
101 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Drsqrt_builder.cc62 kScalarShape, reinterpret_cast<char*>(&kNumeratorMin), in AddNumerator()
65 kScalarShape, reinterpret_cast<char*>(&kNumeratorMax), in AddNumerator()
94 kScalarShape, reinterpret_cast<char*>(&min_value), sizeof(min_value)); in PopulateSubGraph()
96 kScalarShape, reinterpret_cast<char*>(&max_value), sizeof(max_value)); in PopulateSubGraph()
103 auto sqrt_output_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
104 auto sqrt_output_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
127 kScalarShape, reinterpret_cast<char*>(&output_min), sizeof(output_min)); in PopulateSubGraph()
129 kScalarShape, reinterpret_cast<char*>(&output_max), sizeof(output_max)); in PopulateSubGraph()
148 div_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
149 div_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dconv_2d_builder.cc136 kScalarShape, reinterpret_cast<char*>(&data_min), sizeof(data_min)); in PopulateSubGraph()
138 kScalarShape, reinterpret_cast<char*>(&data_max), sizeof(data_max)); in PopulateSubGraph()
207 kScalarShape, reinterpret_cast<char*>(&conv_output_min), in PopulateSubGraph()
210 kScalarShape, reinterpret_cast<char*>(&conv_output_max), in PopulateSubGraph()
266 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
267 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
308 conv_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
309 conv_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
324 batch_to_space_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
326 batch_to_space_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
[all …]
Dreduce_builder.cc61 kScalarShape, reinterpret_cast<char*>(&output_min), sizeof(output_min)); in PopulateSubGraph()
63 kScalarShape, reinterpret_cast<char*>(&output_max), sizeof(output_max)); in PopulateSubGraph()
75 auto mean_out_min = AddOutput(output_element_size, 4, kScalarShape); in PopulateSubGraph()
76 auto mean_out_max = AddOutput(output_element_size, 4, kScalarShape); in PopulateSubGraph()
90 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
91 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dconcat_builder.cc39 kScalarShape, reinterpret_cast<char*>(&concat_axis), sizeof(concat_axis)); in PopulateSubGraph()
67 kScalarShape, reinterpret_cast<char*>(&input_minima_[i]), in PopulateSubGraph()
75 kScalarShape, reinterpret_cast<char*>(&input_maxima_[i]), in PopulateSubGraph()
99 kScalarShape, (char*)&output_min_, sizeof(output_min_)); in PopulateSubGraph()
101 kScalarShape, (char*)&output_max_, sizeof(output_max_)); in PopulateSubGraph()
121 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
122 requantize_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dsquared_difference.cc56 kScalarShape, reinterpret_cast<char*>(&output_min), sizeof(output_min)); in PopulateSubGraph()
58 kScalarShape, reinterpret_cast<char*>(&output_max), sizeof(output_max)); in PopulateSubGraph()
67 auto sub_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
68 auto sub_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
84 mul_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
85 mul_op->AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dtranspose_conv_2d_builder.cc81 kScalarShape, reinterpret_cast<char*>(&weights_min), sizeof(weights_min)); in PopulateSubGraph()
83 kScalarShape, reinterpret_cast<char*>(&weights_max), sizeof(weights_max)); in PopulateSubGraph()
137 kScalarShape, reinterpret_cast<char*>(&zero_bound), sizeof(zero_bound)); in PopulateSubGraph()
139 kScalarShape, reinterpret_cast<char*>(&zero_bound), sizeof(zero_bound)); in PopulateSubGraph()
162 kScalarShape, reinterpret_cast<char*>(&bias_min), sizeof(bias_min)); in PopulateSubGraph()
164 kScalarShape, reinterpret_cast<char*>(&bias_max), sizeof(bias_max)); in PopulateSubGraph()
183 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
184 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dstrided_slice_builder.cc59 kScalarShape, reinterpret_cast<char*>(&begin_mask), sizeof(begin_mask)); in PopulateSubGraph()
62 kScalarShape, reinterpret_cast<char*>(&end_mask), sizeof(end_mask)); in PopulateSubGraph()
65 kScalarShape, reinterpret_cast<char*>(&shrink_axis_mask), in PopulateSubGraph()
80 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
81 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dresize_nearest_neighbor_builder.cc57 kScalarShape, reinterpret_cast<char*>(&align_corners), in PopulateSubGraph()
62 kScalarShape, reinterpret_cast<char*>(&half_pixel_centers), in PopulateSubGraph()
74 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
75 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dresize_bilinear_builder.cc57 kScalarShape, reinterpret_cast<char*>(&align_corners), in PopulateSubGraph()
62 kScalarShape, reinterpret_cast<char*>(&half_pixel_centers), in PopulateSubGraph()
74 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
75 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dspace_to_depth_builder.cc39 kScalarShape, reinterpret_cast<char*>(&block_size_), sizeof(int)); in PopulateSubGraph()
55 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
56 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dsoftmax_builder.cc43 kScalarShape, (char*)&beta_value_, sizeof(beta_value_)); in PopulateSubGraph()
54 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
55 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dsplit_builder.cc51 kScalarShape, reinterpret_cast<char*>(&axis_value), sizeof(int)); in PopulateSubGraph()
70 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
71 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dmatmul_builder.cc202 kScalarShape, reinterpret_cast<char*>(&weights_min_), in PopulateSubGraph()
205 kScalarShape, reinterpret_cast<char*>(&weights_max_), in PopulateSubGraph()
243 kScalarShape, reinterpret_cast<char*>(&weights_min_), in PopulateSubGraph()
246 kScalarShape, reinterpret_cast<char*>(&weights_max_), in PopulateSubGraph()
252 auto transposed_weights_min = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
253 auto transposed_weights_max = AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dl2_normalization_builder.cc45 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
46 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dneg_op_builder.cc39 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
40 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dtranspose_builder.cc44 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
45 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dhardswish_builder.cc48 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
49 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dquantize_builder.cc46 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
47 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dpad_builder.cc60 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
61 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
Dmin_max_builder.cc50 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
51 AddOutput(sizeof(float), 4, kScalarShape); in PopulateSubGraph()
/external/tensorflow/tensorflow/compiler/xla/service/
Dflatten_call_graph_test.cc39 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeScalarComputation()
41 HloInstruction::CreateUnary(kScalarShape, HloOpcode::kNegate, param0)); in MakeScalarComputation()
51 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeMappingComputation()
55 kScalarShape, {last_value}, map_computation)); in MakeMappingComputation()
67 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeCallingComputation()
71 kScalarShape, {last_value}, callee_computation)); in MakeCallingComputation()
81 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeConditionComputation()
96 const Shape kScalarShape = ShapeUtil::MakeShape(F32, {}); member in xla::__anon2cfa76070111::FlattenCallGraphTest
124 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in TEST_F()
126 HloInstruction::CreateCall(kScalarShape, {param0}, c_computation)); in TEST_F()
[all …]
Dcall_graph_test.cc41 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeScalarComputation()
43 HloInstruction::CreateUnary(kScalarShape, opcode, param0)); in MakeScalarComputation()
53 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeMappingComputation()
57 kScalarShape, {last_value}, map_computation)); in MakeMappingComputation()
69 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeCallingComputation()
73 kScalarShape, {last_value}, callee_computation)); in MakeCallingComputation()
83 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in MakeConditionComputation()
92 const Shape kScalarShape = ShapeUtil::MakeShape(F32, {}); member in xla::__anon622e181a0111::CallGraphTest
211 HloInstruction::CreateParameter(0, kScalarShape, "param0")); in TEST_F()
213 HloInstruction::CreateCall(kScalarShape, {param0}, subcomputation)); in TEST_F()
[all …]

12