Searched refs:F32 (Results 1 - 25 of 196) sorted by relevance

12345678

/external/vixl/test/aarch32/
H A Dtest-assembler-cond-dt-drt-drd-drn-drm-float-f32-only-a32.cc106 {{{F32, d1, d2, d30}, false, al, "F32 d1 d2 d30", "F32_d1_d2_d30"},
107 {{F32, d12, d5, d13}, false, al, "F32 d12 d5 d13", "F32_d12_d5_d13"},
108 {{F32, d3, d16, d2}, false, al, "F32 d3 d16 d2", "F32_d3_d16_d2"},
109 {{F32, d21, d26, d7}, false, al, "F32 d21 d26 d7", "F32_d21_d26_d7"},
110 {{F32, d30, d1, d16}, false, al, "F32 d3
[all...]
H A Dtest-assembler-cond-dt-drt-drd-drn-drm-float-f32-only-t32.cc106 {{{F32, d1, d2, d30}, false, al, "F32 d1 d2 d30", "F32_d1_d2_d30"},
107 {{F32, d12, d5, d13}, false, al, "F32 d12 d5 d13", "F32_d12_d5_d13"},
108 {{F32, d3, d16, d2}, false, al, "F32 d3 d16 d2", "F32_d3_d16_d2"},
109 {{F32, d21, d26, d7}, false, al, "F32 d21 d26 d7", "F32_d21_d26_d7"},
110 {{F32, d30, d1, d16}, false, al, "F32 d3
[all...]
H A Dtest-assembler-cond-dt-drt-drd-drn-drm-float-not-f16-a32.cc100 {{F32, d5, d16, d12}, false, al, "F32 d5 d16 d12", "F32_d5_d16_d12"},
103 {{F32, d22, d28, d12}, false, al, "F32 d22 d28 d12", "F32_d22_d28_d12"},
104 {{F32, d15, d1, d6}, false, al, "F32 d15 d1 d6", "F32_d15_d1_d6"},
105 {{F32, d18, d17, d23}, false, al, "F32 d18 d17 d23", "F32_d18_d17_d23"},
112 {{F32, d4, d26, d15}, false, al, "F32 d
[all...]
H A Dtest-assembler-cond-dt-drt-drd-drn-drm-float-not-f16-t32.cc100 {{F32, d5, d16, d12}, false, al, "F32 d5 d16 d12", "F32_d5_d16_d12"},
103 {{F32, d22, d28, d12}, false, al, "F32 d22 d28 d12", "F32_d22_d28_d12"},
104 {{F32, d15, d1, d6}, false, al, "F32 d15 d1 d6", "F32_d15_d1_d6"},
105 {{F32, d18, d17, d23}, false, al, "F32 d18 d17 d23", "F32_d18_d17_d23"},
112 {{F32, d4, d26, d15}, false, al, "F32 d
[all...]
/external/tensorflow/tensorflow/compiler/xla/
H A Dlayout_util_test.cc48 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 3}, {0, 1})});
50 ShapeUtil::MakeTupleShape({MakeShapeWithLayout(F32, {2, 2}, {0, 1})});
73 Shape src = MakeShapeWithLayout(F32, {2, 3}, {0, 1});
74 Shape dst = MakeShapeWithLayout(F32, {2, 3}, {1, 0});
96 Shape src = MakeShapeWithSparseLayout(F32, {2, 3}, 2);
97 Shape dst = MakeShapeWithLayout(F32, {2, 3}, {1, 0});
120 {MakeShapeWithLayout(F32, {2, 3}, {0, 1}),
121 MakeShapeWithLayout(F32, {42, 123}, {1, 0}),
123 {MakeShapeWithLayout(F32, {}, {}),
124 MakeShapeWithLayout(F32, {
[all...]
H A Dshape_util_test.cc32 Shape matrix = ShapeUtil::MakeShape(F32, {2, 3});
38 auto shape = ShapeUtil::MakeShape(F32, {1, 2, 3, 4});
43 Shape matrix = ShapeUtil::MakeShape(F32, {2, 3});
48 Shape shape = ShapeUtil::MakeShape(F32, {3});
53 Shape shape = ShapeUtil::MakeShape(F32, {3, 2});
59 Shape shape = ShapeUtil::MakeShape(F32, {3, 2, 7});
66 Shape shape = ShapeUtil::MakeShape(F32, {3, 2, 7, 8});
77 Shape expected = ShapeUtil::MakeShape(F32, {123, 456});
88 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {1572864}),
100 ShapeUtil::MakeShape(F32, {
[all...]
H A Dprimitive_util.cc24 return type == F16 || type == F32 || type == F64 || type == BF16;
58 case F32:
81 return F32;
H A Dindex_util_test.cc38 Shape vector_shape = ShapeUtil::MakeShape(F32, {100});
49 Shape matrix_shape_01 = ShapeUtil::MakeShape(F32, {10, 20});
65 Shape matrix_shape_10 = ShapeUtil::MakeShape(F32, {10, 20});
81 Shape shape_210 = ShapeUtil::MakeShape(F32, {10, 20, 30});
94 Shape shape_120 = ShapeUtil::MakeShape(F32, {10, 20, 30});
107 Shape shape_3210 = ShapeUtil::MakeShape(F32, {10, 20, 30, 40});
131 Shape shape = ShapeUtil::MakeShape(F32, {10, 20, 30, 40, 30, 20, 10});
/external/tensorflow/tensorflow/compiler/xla/service/
H A Dbatchnorm_expander_test.cc43 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 2, 2, 2});
44 Shape scale_shape = ShapeUtil::MakeShape(F32, {2});
45 Shape offset_shape = ShapeUtil::MakeShape(F32, {2});
77 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 2, 2, 2});
78 Shape scale_shape = ShapeUtil::MakeShape(F32, {2});
79 Shape mean_shape = ShapeUtil::MakeShape(F32, {2});
80 Shape var_shape = ShapeUtil::MakeShape(F32, {2});
81 Shape grad_output_shape = ShapeUtil::MakeShape(F32, {2, 2, 2, 2});
H A Dhlo_cost_analysis_test.cc66 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
77 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
78 auto y = builder.Parameter(1, ShapeUtil::MakeShape(F32, {}), "y");
88 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
99 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
100 auto y = builder.Parameter(1, ShapeUtil::MakeShape(F32, {}), "y");
110 auto x = builder.Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
111 auto y = builder.Parameter(1, ShapeUtil::MakeShape(F32, {}), "y");
149 auto lhs = builder.Parameter(0, ShapeUtil::MakeShape(F32, {10, 5}), "lhs");
150 auto rhs = builder.Parameter(1, ShapeUtil::MakeShape(F32, {
[all...]
H A Dbfloat16_normalization_test.cc79 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4});
102 EXPECT_EQ(add1->shape().element_type(), F32);
107 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4});
130 EXPECT_EQ(mul0->shape().element_type(), F32);
131 EXPECT_EQ(mul1->shape().element_type(), F32);
137 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4});
160 EXPECT_EQ(sub0->shape().element_type(), F32);
161 EXPECT_EQ(sub1->shape().element_type(), F32);
166 Shape f32_input_shape = ShapeUtil::MakeShape(F32, {2, 4});
167 Shape f32_output_shape = ShapeUtil::MakeShape(F32, {
[all...]
H A Dreshape_mover_test.cc41 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7});
43 0, ShapeUtil::MakeShape(F32, {1, 8, 1, 7}), "param0"));
45 1, ShapeUtil::MakeShape(F32, {1, 8, 7, 1}), "param1"));
78 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7});
80 HloInstruction::CreateRng(ShapeUtil::MakeShape(F32, {1, 8, 1, 7, 1}),
104 auto root_shape = ShapeUtil::MakeShape(F32, {});
106 0, ShapeUtil::MakeShape(F32, {1, 1, 1}), "param0"));
108 1, ShapeUtil::MakeShape(F32, {1, 1, 1}), "param1"));
130 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7});
132 0, ShapeUtil::MakeShape(F32, {
[all...]
H A Dhlo_element_type_converter_test.cc45 HloElementTypeConverter type_converter(BF16, F32);
59 HloElementTypeConverter type_converter(BF16, F32);
79 HloElementTypeConverter type_converter(BF16, F32);
107 HloElementTypeConverter type_converter(BF16, F32);
H A Dshape_inference_test.cc38 const Shape f32_ = ShapeUtil::MakeShape(F32, {});
42 // Some handy vector and matrix shapes of F32 type.
44 const Shape vector_32_ = ShapeUtil::MakeShape(F32, {32});
45 const Shape vector_64_ = ShapeUtil::MakeShape(F32, {64});
46 const Shape matrix_32_48_ = ShapeUtil::MakeShape(F32, {32, 48});
47 const Shape matrix_32_64_ = ShapeUtil::MakeShape(F32, {32, 64});
48 const Shape matrix_64_48_ = ShapeUtil::MakeShape(F32, {64, 48});
59 // element type here is hard-coded to F32.
76 operand_shape_ = ShapeUtil::MakeShape(F32, {8, 16});
77 source_shape_ = ShapeUtil::MakeShape(F32, {
[all...]
H A Dimplicit_broadcast_remover_test.cc36 const Shape shape = ShapeUtil::MakeShape(F32, {2, 4});
55 const Shape shape = ShapeUtil::MakeShape(F32, {2, 4});
57 0, ShapeUtil::MakeShape(F32, {}), "scalar_param"));
81 const Shape shape = ShapeUtil::MakeShape(F32, {2, 4, 6});
85 1, ShapeUtil::MakeShape(F32, {1, 4, 1}), "p1"));
103 const Shape shape = ShapeUtil::MakeShape(F32, {1, 4, 1});
105 0, ShapeUtil::MakeShape(F32, {}), "scalar_param"));
125 const Shape shape = ShapeUtil::MakeShape(F32, {2, 4, 6, 8});
127 0, ShapeUtil::MakeShape(F32, {1, 4, 1, 8}), "p0"));
129 1, ShapeUtil::MakeShape(F32, {
[all...]
H A Dhlo_dce_test.cc98 0, ShapeUtil::MakeShape(F32, {}), "live_param"));
100 1, ShapeUtil::MakeShape(F32, {}), "dead_param1"));
102 2, ShapeUtil::MakeShape(F32, {}), "dead_param2"));
178 Shape shape = ShapeUtil::MakeShape(F32, {});
218 Shape shape = ShapeUtil::MakeShape(F32, {});
273 Shape shape = ShapeUtil::MakeShape(F32, {});
330 /*parameter_number=*/0, ShapeUtil::MakeShape(F32, {}), "param0"));
333 /*parameter_number=*/1, ShapeUtil::MakeShape(F32, {}), "param1"));
335 ShapeUtil::MakeShape(F32, {}), HloOpcode::kAdd, param0, param1));
341 ShapeUtil::MakeShape(F32, {
[all...]
H A Dalgebraic_simplifier_test.cc55 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
76 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
97 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
122 Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 2});
144 Shape r2f32 = ShapeUtil::MakeShape(F32, {3, 2});
167 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
188 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
209 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
237 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
265 Shape r0f32 = ShapeUtil::MakeShape(F32, {});
[all...]
/external/vixl/examples/aarch32/
H A Dmandelbrot.cc43 __ Vmov(F32, kCRealStep, 0.125);
44 __ Vmov(F32, kCImagStep, 0.0625);
62 __ Vmov(F32, kModSqLimit, 4.0);
119 __ Vmov(F32, kZReal, 0.0);
122 __ Vmov(F32, kZImag, 0.0);
124 __ Vmov(F32, kFlags, 0.0);
135 __ Vmul(F32, q10, kZReal, kZImag); // re(zn) * im(zn)
137 __ Vmul(F32, kZReal, kZReal, kZReal); // re(zn)^2
138 __ Vadd(F32, kZReal, kCReal, kZReal); // re(c) + re(zn)^2
139 __ Vmls(F32, kZRea
[all...]
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dhlo_metadata_test.cc34 auto x = builder->Parameter(0, ShapeUtil::MakeShape(F32, {}), "x");
35 auto y = builder->Parameter(1, ShapeUtil::MakeShape(F32, {}), "y");
48 Shape argument_layout = ShapeUtil::MakeShape(F32, {});
70 Shape argument_layout = ShapeUtil::MakeShape(F32, {});
H A Dquery_inferred_shape_test.cc39 ASSERT_TRUE(ShapeUtil::Equal(*shape, ShapeUtil::MakeShape(F32, {})));
H A Dconvert_test.cc56 builder.ConvertElementType(a, F32);
65 builder.ConvertElementType(a, F32);
83 builder.ConvertElementType(a, F32);
92 builder.ConvertElementType(a, F32);
110 builder.ConvertElementType(a, F32);
119 builder.ConvertElementType(a, F32);
155 builder.ConvertElementType(a, F32);
165 builder.ConvertElementType(a, F32);
176 auto param = b->Parameter(0, ShapeUtil::MakeShape(F32, {}), "in");
189 b->ConvertElementType(param, F32);
[all...]
H A Dvector_ops_reduce_test.cc57 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
69 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
108 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
125 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
140 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
152 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
165 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
177 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
187 auto sum_reducer = CreateScalarAddComputation(F32, &builder_);
197 auto sum_reducer = CreateScalarAddComputation(F32,
[all...]
H A Dbitcast_convert_test.cc56 builder.BitcastConvertType(a, F32);
68 builder.BitcastConvertType(a, F32);
77 builder.BitcastConvertType(a, F32);
96 builder.BitcastConvertType(a, F32);
105 auto param = b->Parameter(0, ShapeUtil::MakeShape(F32, {}), "in");
118 b->BitcastConvertType(param, F32);
135 builder.BitcastConvertType(reshape, F32);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc45 0, ShapeUtil::MakeShape(F32, {1024, 256}), "arg0"));
47 1, ShapeUtil::MakeShape(F32, {256, 1}), "arg1"));
52 MakeDot(ShapeUtil::MakeShape(F32, {1024, 1}), exp0, arg1));
64 0, ShapeUtil::MakeShape(F32, {1, 256}), "arg0"));
66 1, ShapeUtil::MakeShape(F32, {256, 1024}), "arg1"));
71 MakeDot(ShapeUtil::MakeShape(F32, {1, 1024}), arg0, exp1));
83 0, ShapeUtil::MakeShape(F32, {2, 512, 2, 128}), "arg0"));
85 1, ShapeUtil::MakeShape(F32, {256, 1}), "arg1"));
92 MakeDot(ShapeUtil::MakeShape(F32, {1024, 1}), bitcast0, arg1));
104 0, ShapeUtil::MakeShape(F32, {
[all...]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_convolution_rewriter_test.cc99 0, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "activations"));
102 1, ShapeUtil::MakeShape(F32, {1, 1, 2, 1}), "gradients"));
128 0, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "activations"));
131 1, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "gradients"));
157 0, ShapeUtil::MakeShape(F32, {20, 35, 35, 32}), "activations"));
160 1, ShapeUtil::MakeShape(F32, {20, 35, 35, 32}), "gradients"));
169 ShapeUtil::MakeShape(F32, {32, 3, 3, 32}), activations, gradients,
187 0, ShapeUtil::MakeShape(F32, {20, 10, 10, 192}), "activations"));
190 1, ShapeUtil::MakeShape(F32, {20, 4, 4, 320}), "gradients"));
199 ShapeUtil::MakeShape(F32, {32
[all...]

Completed in 3438 milliseconds

12345678