/external/vixl/test/aarch32/ |
H A D | test-assembler-cond-dt-drt-drd-drn-drm-float-f32-only-a32.cc | 106 {{{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 D | test-assembler-cond-dt-drt-drd-drn-drm-float-f32-only-t32.cc | 106 {{{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 D | test-assembler-cond-dt-drt-drd-drn-drm-float-not-f16-a32.cc | 100 {{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 D | test-assembler-cond-dt-drt-drd-drn-drm-float-not-f16-t32.cc | 100 {{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 D | layout_util_test.cc | 48 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 D | shape_util_test.cc | 32 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 D | primitive_util.cc | 24 return type == F16 || type == F32 || type == F64 || type == BF16; 58 case F32: 81 return F32;
|
H A D | index_util_test.cc | 38 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 D | batchnorm_expander_test.cc | 43 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 D | hlo_cost_analysis_test.cc | 66 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 D | bfloat16_normalization_test.cc | 79 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 D | reshape_mover_test.cc | 41 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 D | hlo_element_type_converter_test.cc | 45 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 D | shape_inference_test.cc | 38 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 D | implicit_broadcast_remover_test.cc | 36 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 D | hlo_dce_test.cc | 98 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 D | algebraic_simplifier_test.cc | 55 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 D | mandelbrot.cc | 43 __ 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 D | hlo_metadata_test.cc | 34 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 D | query_inferred_shape_test.cc | 39 ASSERT_TRUE(ShapeUtil::Equal(*shape, ShapeUtil::MakeShape(F32, {})));
|
H A D | convert_test.cc | 56 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 D | vector_ops_reduce_test.cc | 57 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 D | bitcast_convert_test.cc | 56 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 D | cpu_instruction_fusion_test.cc | 45 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 D | cudnn_convolution_rewriter_test.cc | 99 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...] |