| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | symbolic_shapes_test.cc | 32 TensorShapeProto MakeShape(std::vector<int> dims) { in MakeShape() function in tensorflow::grappler::__anon3a6d31fa0111::SymbolicShapesTest
49 EXPECT_FALSE(ShapeIsSymbolicallyDefined(MakeShape({-1, 2}))); in TEST_F()
51 EXPECT_TRUE(ShapeIsSymbolicallyDefined(MakeShape({1, 2}))); in TEST_F()
52 EXPECT_TRUE(ShapeIsSymbolicallyDefined(MakeShape({-2, 2}))); in TEST_F()
57 EXPECT_FALSE(ShapesSymbolicallyEqual(MakeShape({-1, 2}), MakeShape({-1, 2}))); in TEST_F()
58 EXPECT_FALSE(ShapesSymbolicallyEqual(MakeShape({-2, 2}), MakeShape({-3, 2}))); in TEST_F()
60 EXPECT_TRUE(ShapesSymbolicallyEqual(MakeShape({1, 2}), MakeShape({1, 2}))); in TEST_F()
61 EXPECT_TRUE(ShapesSymbolicallyEqual(MakeShape({-2, 2}), MakeShape({-2, 2}))); in TEST_F()
66 EXPECT_FALSE(ShapesBroadcastable(MakeShape({-2}), MakeShape({1, -3}))); in TEST_F()
67 EXPECT_FALSE(ShapesBroadcastable(MakeShape({-1, 2}), MakeShape({-1, 2}))); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util_test.cc | 44 Shape matrix = ShapeUtil::MakeShape(F32, {2, 3}); in TEST()
50 auto shape = ShapeUtil::MakeShape(F32, {1, 2, 3, 4}); in TEST()
55 Shape matrix = ShapeUtil::MakeShape(F32, {2, 3}); in TEST()
60 Shape shape = ShapeUtil::MakeShape(F32, {3}); in TEST()
65 Shape shape = ShapeUtil::MakeShape(F32, {3, 2}); in TEST()
71 Shape shape = ShapeUtil::MakeShape(F32, {3, 2, 7}); in TEST()
78 Shape shape = ShapeUtil::MakeShape(F32, {3, 2, 7, 8}); in TEST()
86 Shape shape1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST()
87 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST()
95 ShapeUtil::MakeShape(F32, {}))); in TEST()
[all …]
|
| D | index_util_test.cc | 37 Shape vector_shape = ShapeUtil::MakeShape(F32, {100}); in TEST()
48 Shape matrix_shape_01 = ShapeUtil::MakeShape(F32, {10, 20}); in TEST()
64 Shape matrix_shape_10 = ShapeUtil::MakeShape(F32, {10, 20}); in TEST()
80 Shape shape_210 = ShapeUtil::MakeShape(F32, {10, 20, 30}); in TEST()
93 Shape shape_120 = ShapeUtil::MakeShape(F32, {10, 20, 30}); in TEST()
106 Shape shape_3210 = ShapeUtil::MakeShape(F32, {10, 20, 30, 40}); in TEST()
130 Shape shape = ShapeUtil::MakeShape(F32, {10, 20, 30, 40, 30, 20, 10}); in TEST()
142 auto shape = ShapeUtil::MakeShape(S32, {2, 2}); in TEST()
|
| D | shape_test.cc | 37 const Shape scalar_ = ShapeUtil::MakeShape(F32, {});
40 const Shape matrix_ = ShapeUtil::MakeShape(U32, {1, 2});
47 ShapeUtil::MakeShape(S32, {5, 2}, {true, false});
85 ShapeUtil::MakeShape(F32, {23, 44, 55}, {true, false, true}); in TEST_F()
144 *program_shape.add_parameters() = ShapeUtil::MakeShape(F32, {1, 2, 3}); in TEST_F()
146 *program_shape.add_parameters() = ShapeUtil::MakeShape(S64, {}); in TEST_F()
148 {ShapeUtil::MakeShape(S32, {}), in TEST_F()
150 ShapeUtil::MakeShape(F32, {42, 42})}); in TEST_F()
152 *program_shape.mutable_result() = ShapeUtil::MakeShape(F32, {7}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 140 Shape s = ShapeUtil::MakeShape(PRED, {}); in TEST_F()
156 Shape s = ShapeUtil::MakeShape(PRED, {}); in TEST_F()
182 auto x = Parameter(&b, 0, ShapeUtil::MakeShape(F32, {3, 5}), "x"); in TEST_F()
191 const auto& x_shape = ShapeUtil::MakeShape(S32, {2, 4, 6}); in TEST_F()
192 const auto& y_shape = ShapeUtil::MakeShape(S32, {2, 4}); in TEST_F()
207 auto x = Parameter(&b, 0, ShapeUtil::MakeShape(S32, {1, 3, 5, 7}), "x"); in TEST_F()
216 auto x = Parameter(&b, 0, ShapeUtil::MakeShape(U32, {2, 4, 6}), "x"); in TEST_F()
217 auto y = Parameter(&b, 1, ShapeUtil::MakeShape(U32, {2, 4}), "y"); in TEST_F()
226 Parameter(&b_call, 0, ShapeUtil::MakeShape(PRED, {}), "x"); in TEST_F()
229 auto x = Parameter(&b, 0, ShapeUtil::MakeShape(PRED, {}), "x"); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 37 const Shape s32_ = ShapeUtil::MakeShape(S32, {});
38 const Shape f16_ = ShapeUtil::MakeShape(F16, {});
39 const Shape f32_ = ShapeUtil::MakeShape(F32, {});
40 const Shape f64_ = ShapeUtil::MakeShape(F64, {});
41 const Shape pred_ = ShapeUtil::MakeShape(PRED, {});
45 const Shape vector_32_ = ShapeUtil::MakeShape(F32, {32});
46 const Shape vector_64_ = ShapeUtil::MakeShape(F32, {64});
47 const Shape matrix_32_48_ = ShapeUtil::MakeShape(F32, {32, 48});
48 const Shape matrix_32_64_ = ShapeUtil::MakeShape(F32, {32, 64});
49 const Shape matrix_64_48_ = ShapeUtil::MakeShape(F32, {64, 48});
[all …]
|
| D | hlo_cost_analysis_test.cc | 63 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
74 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
75 auto y = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {}), "y"); in HloCostAnalysisTest()
85 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
96 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
97 auto y = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {}), "y"); in HloCostAnalysisTest()
107 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
108 auto y = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {}), "y"); in HloCostAnalysisTest()
141 auto lhs = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {10, 5}), "lhs"); in TEST_F()
142 auto rhs = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {5, 30}), "rhs"); in TEST_F()
[all …]
|
| D | hlo_sharding_test.cc | 51 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4}), in TEST_F()
67 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4}), in TEST_F()
70 sharding.Validate(ShapeUtil::MakeShape(U32, {4}), /*num_devices=*/5)); in TEST_F()
73 sharding.GetAsShapeTree(ShapeUtil::MakeShape(U32, {4})); in TEST_F()
82 EXPECT_IS_NOT_OK(sharding.Validate(ShapeUtil::MakeShape(F32, {4, 6}), in TEST_F()
89 EXPECT_IS_NOT_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4, 6}), in TEST_F()
95 Shape shape = ShapeUtil::MakeShape(U32, {4, 5}); in TEST_F()
97 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(F32, {3, 5}), in TEST_F()
128 ShapeUtil::MakeShape(F32, {}), in TEST_F()
129 ShapeUtil::MakeTupleShape({ShapeUtil::MakeShape(F32, {3})}), in TEST_F()
[all …]
|
| D | reshape_mover_test.cc | 42 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F()
44 0, ShapeUtil::MakeShape(F32, {1, 8, 1, 7}), "param0")); in TEST_F()
46 1, ShapeUtil::MakeShape(F32, {1, 8, 7, 1}), "param1")); in TEST_F()
80 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7});
82 ShapeUtil::MakeShape(F32, {1, 8, 1, 7, 1}),
111 auto root_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
113 0, ShapeUtil::MakeShape(F32, {1, 1, 1}), "param0")); in TEST_F()
115 1, ShapeUtil::MakeShape(F32, {1, 1, 1}), "param1")); in TEST_F()
138 auto root_shape = ShapeUtil::MakeShape(F32, {8, 7}); in TEST_F()
140 0, ShapeUtil::MakeShape(F32, {1, 8, 1, 7}), "param0")); in TEST_F()
[all …]
|
| D | dynamic_dimension_inference_test.cc | 56 0, ShapeUtil::MakeShape(F32, {}), "lhs")); in GetAdd()
58 1, ShapeUtil::MakeShape(F32, {}), "rhs")); in GetAdd()
67 0, ShapeUtil::MakeShape(F32, {}), "lhs")); in GetGe()
69 1, ShapeUtil::MakeShape(F32, {}), "rhs")); in GetGe()
71 ShapeUtil::MakeShape(PRED, {}), lhs, rhs, ComparisonDirection::kGe)); in GetGe()
77 const Shape scalar_shape_ = ShapeUtil::MakeShape(S32, {});
82 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F()
105 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F()
128 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F()
157 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F()
[all …]
|
| D | bfloat16_propagation_test.cc | 100 Shape shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
113 ShapeUtil::MakeShape(PRED, {2, 4}), a, b, ComparisonDirection::kEq)); in TEST_F()
118 ShapeUtil::MakeShape(F32, {4, 2}), sel, {1, 0})); in TEST_F()
120 CreateDot(ShapeUtil::MakeShape(F32, {4, 4}), xpose, a)); in TEST_F()
122 ShapeUtil::MakeShape(F32, {4, 4}), HloOpcode::kAdd, dot, dot)); in TEST_F()
144 HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {}), "a")); in TEST_F()
146 HloInstruction::CreateParameter(1, ShapeUtil::MakeShape(F32, {}), "b")); in TEST_F()
148 ShapeUtil::MakeShape(F32, {}), HloOpcode::kMaximum, p0, p1)); in TEST_F()
152 Shape shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
179 ShapeUtil::MakeShape(F32, {4, 2}), c, {1, 0})); in TEST_F()
[all …]
|
| D | dynamic_padder_test.cc | 59 0, ShapeUtil::MakeShape(F32, {}), "lhs")); in GetScalarAddComputation()
61 1, ShapeUtil::MakeShape(F32, {}), "rhs")); in GetScalarAddComputation()
68 const Shape scalar_shape_ = ShapeUtil::MakeShape(U32, {});
73 auto input_shape = ShapeUtil::MakeShape(F32, {1, 2, 2}); in TEST_F()
74 auto reduce_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F()
107 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
108 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()
109 auto zx_shape = ShapeUtil::MakeShape(F32, {zdim, xdim}); in TEST_F()
150 auto xy_shape = ShapeUtil::MakeShape(F32, {xdim, ydim}); in TEST_F()
151 auto yz_shape = ShapeUtil::MakeShape(F32, {ydim, zdim}); in TEST_F()
[all …]
|
| D | algebraic_simplifier_test.cc | 58 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F()
281 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F()
300 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F()
322 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F()
344 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F()
372 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
381 Shape r4f32 = ShapeUtil::MakeShape(F32, {4, 5, 6, 7}); in TEST_F()
385 Shape r3f32 = ShapeUtil::MakeShape(F32, {5, 6, 7}); in TEST_F()
389 Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F()
403 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F()
[all …]
|
| D | hlo_dce_test.cc | 97 0, ShapeUtil::MakeShape(F32, {}), "live_param")); in TEST_F()
99 1, ShapeUtil::MakeShape(F32, {}), "dead_param1")); in TEST_F()
101 2, ShapeUtil::MakeShape(F32, {}), "dead_param2")); in TEST_F()
177 Shape shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
217 Shape shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
227 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), param, in TEST_F()
273 Shape shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
332 /*parameter_number=*/0, ShapeUtil::MakeShape(F32, {}), "param0")); in TEST_F()
335 /*parameter_number=*/1, ShapeUtil::MakeShape(F32, {}), "param1")); in TEST_F()
337 ShapeUtil::MakeShape(F32, {}), HloOpcode::kAdd, param0, param1)); in TEST_F()
[all …]
|
| D | bfloat16_normalization_test.cc | 94 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
95 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
122 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
123 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
152 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
153 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
181 Shape f32_input_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
182 Shape f32_output_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F()
184 Shape bf16_scalar_shape = ShapeUtil::MakeShape(BF16, {}); in TEST_F()
240 /*parameter_number=*/0, ShapeUtil::MakeShape(F32, {}), "x")); in TEST_F()
[all …]
|
| D | batchnorm_expander_test.cc | 57 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 2, 2, 2}); in TEST_F()
58 Shape scale_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F()
59 Shape offset_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F()
92 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 2, 2, 2}); in TEST_F()
93 Shape scale_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F()
94 Shape mean_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F()
95 Shape var_shape = ShapeUtil::MakeShape(F32, {2}); in TEST_F()
96 Shape grad_output_shape = ShapeUtil::MakeShape(F32, {2, 2, 2, 2}); in TEST_F()
|
| D | hlo_matchers_test.cc | 43 auto shape = ShapeUtil::MakeShape(F32, {1}); in TEST()
84 ShapeUtil::MakeShape(F32, {1}), {c1.get(), c2.get()}, "foo_target"); in TEST()
111 EXPECT_THAT(p0.get(), op::Shape(ShapeUtil::MakeShape(F32, {5, 7}))); in TEST()
115 ::testing::Not(op::ShapeWithLayout(ShapeUtil::MakeShape(F32, {5, 7})))); in TEST()
118 ::testing::Not(op::Shape(ShapeUtil::MakeShape(F32, {7, 5})))); in TEST()
122 ::testing::Not(op::ShapeWithLayout(ShapeUtil::MakeShape(F32, {7, 5})))); in TEST()
135 EXPECT_THAT(Explain(p0.get(), op::Shape(ShapeUtil::MakeShape(F32, {7, 5}))), in TEST()
146 auto p0 = HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {5}), in TEST()
149 auto p1 = HloInstruction::CreateParameter(1, ShapeUtil::MakeShape(F32, {7}), in TEST()
154 {ShapeUtil::MakeShape(F32, {7}), ShapeUtil::MakeShape(S32, {9}), in TEST()
[all …]
|
| D | hlo_instruction_test.cc | 45 Shape r0f32_ = ShapeUtil::MakeShape(F32, {});
329 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F()
330 Shape f32a100x10 = ShapeUtil::MakeShape(F32, {100, 10}); in TEST_F()
367 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F()
368 Shape f32v100 = ShapeUtil::MakeShape(F32, {100}); in TEST_F()
369 Shape f32a100x10 = ShapeUtil::MakeShape(F32, {100, 10}); in TEST_F()
752 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
910 Shape bcast_shape1 = ShapeUtil::MakeShape(F32, {2, 2, 42}); in TEST_F()
911 Shape bcast_shape2 = ShapeUtil::MakeShape(F32, {2, 2, 123}); in TEST_F()
971 const Shape f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 52 0, ShapeUtil::MakeShape(F32, {1024, 256}), "arg0")); in TEST_F()
54 1, ShapeUtil::MakeShape(F32, {256, 1}), "arg1")); in TEST_F()
57 ShapeUtil::MakeShape(S32, {1024, 256}), HloOpcode::kExp, arg0)); in TEST_F()
59 MakeDot(ShapeUtil::MakeShape(F32, {1024, 1}), exp0, arg1)); in TEST_F()
71 0, ShapeUtil::MakeShape(F32, {1, 256}), "arg0")); in TEST_F()
73 1, ShapeUtil::MakeShape(F32, {256, 1024}), "arg1")); in TEST_F()
76 ShapeUtil::MakeShape(S32, {256, 1024}), HloOpcode::kExp, arg1)); in TEST_F()
78 MakeDot(ShapeUtil::MakeShape(F32, {1, 1024}), arg0, exp1)); in TEST_F()
90 0, ShapeUtil::MakeShape(F32, {2, 512, 2, 128}), "arg0")); in TEST_F()
92 1, ShapeUtil::MakeShape(F32, {256, 1}), "arg1")); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_rewriter_test.cc | 102 0, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "activations")); in TEST_F()
105 1, ShapeUtil::MakeShape(F32, {1, 1, 2, 1}), "gradients")); in TEST_F()
143 0, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "activations")); in TEST_F()
146 1, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "gradients")); in TEST_F()
173 0, ShapeUtil::MakeShape(F32, {20, 35, 35, 32}), "activations")); in TEST_F()
176 1, ShapeUtil::MakeShape(F32, {20, 35, 35, 32}), "gradients")); in TEST_F()
185 ShapeUtil::MakeShape(F32, {32, 3, 3, 32}), activations, gradients, in TEST_F()
203 0, ShapeUtil::MakeShape(F32, {20, 10, 10, 192}), "activations")); in TEST_F()
206 1, ShapeUtil::MakeShape(F32, {20, 4, 4, 320}), "gradients")); in TEST_F()
215 ShapeUtil::MakeShape(F32, {320, 3, 3, 192}), activations, gradients, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | fusion_test.cc | 102 ShapeUtil::MakeShape(prim_type, {test_width, test_height}); in TestElementwise2D()
208 ShapeUtil::MakeShape(F32, {3, 1}), HloOpcode::kAdd, const0, const1)); in XLA_TEST_F()
210 ShapeUtil::MakeShape(F32, {1, 3}), add2, {1, 0})); in XLA_TEST_F()
214 ShapeUtil::MakeShape(F32, {2, 3}), {reshape3, const4}, 0)); in XLA_TEST_F()
218 ShapeUtil::MakeShape(F32, {2, 3}), HloOpcode::kNegate, const6)); in XLA_TEST_F()
220 ShapeUtil::MakeShape(F32, {2, 3}), HloOpcode::kAdd, concat5, negate7)); in XLA_TEST_F()
227 HloInstruction::CreateTernary(ShapeUtil::MakeShape(F32, {2, 3}), in XLA_TEST_F()
230 ShapeUtil::MakeShape(F32, {2, 1}), select11, {0, 1}, {2, 2}, {1, 1})); in XLA_TEST_F()
254 ShapeUtil::MakeShape(F32, {1, 3}), HloOpcode::kCopy, const0)); in XLA_TEST_F()
259 ShapeUtil::MakeShape(F32, {1, 3}), HloOpcode::kAdd, copy1, const2)); in XLA_TEST_F()
[all …]
|
| D | while_test.cc | 52 auto result_shape = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F()
88 auto result_shape = ShapeUtil::MakeShape(S64, {}); in XLA_TEST_F()
118 auto result_shape = ShapeUtil::MakeShape(S32, {}); in XLA_TEST_F()
119 auto orig_shape = ShapeUtil::MakeShape(S32, {2}); in XLA_TEST_F()
151 auto result_shape = ShapeUtil::MakeShape(PRED, {}); in XLA_TEST_F()
188 Shape result_shape = ShapeUtil::MakeShape(F32, {0}); in XLA_TEST_F()
194 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in XLA_TEST_F()
195 auto y = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {}), "y"); in XLA_TEST_F()
242 Shape result_shape = ShapeUtil::MakeShape(F32, {8}); in XLA_TEST_F()
248 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in XLA_TEST_F()
[all …]
|
| D | broadcast_test.cc | 42 ShapeUtil::MakeShape(F32, {}), input, {})); in XLA_TEST_F()
58 ShapeUtil::MakeShape(F32, {2, 2}), input, {})); in XLA_TEST_F()
78 ShapeUtil::MakeShape(F32, {3, 2}), input, {0})); in XLA_TEST_F()
80 ShapeUtil::MakeShape(F32, {2, 3}), input, {1})); in XLA_TEST_F()
102 ShapeUtil::MakeShape(F32, {2, 2}), input, {0, 1})); in XLA_TEST_F()
121 ShapeUtil::MakeShape(F32, {2, 2}), input, {1, 0})); in XLA_TEST_F()
138 ShapeUtil::MakeShape(F32, {2, 3, 2}), input, {0, 2})); in XLA_TEST_F()
158 ShapeUtil::MakeShape(F32, {2, 2, 3, 3}), input, {1})); in TEST_F()
183 ShapeUtil::MakeShape(F32, {3, 3, 3, r1_size}), input, {3})); in TEST_F()
214 ShapeUtil::MakeShape(F32, {32, 64, 7, 7}), input, {1})); in XLA_TEST_F()
[all …]
|
| D | map_test.cc | 55 auto x = Parameter(&mapped_builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in CreateAdderToOne()
65 auto lhs = Parameter(&b, 0, ShapeUtil::MakeShape(F32, {}), "x"); in CreateMax()
66 auto rhs = Parameter(&b, 1, ShapeUtil::MakeShape(F32, {}), "y"); in CreateMax()
78 (void)Parameter(&mapped_builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in CreateScalarOne()
92 auto x = Parameter(&mapped_builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in CreateMulByTwo()
110 auto x = Parameter(&mapped_builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in CreateAdderToOneTimesItself()
128 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in CreateMapPlusN()
141 auto x = Parameter(&b, 0, ShapeUtil::MakeShape(F32, {}), "x"); in CreateGt()
142 auto y = Parameter(&b, 1, ShapeUtil::MakeShape(F32, {}), "y"); in CreateGt()
158 auto x = Parameter(&mapped_builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in CreateTernaryAdder()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.cc | 86 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateMax()
88 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateMax()
102 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateMin()
104 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateMin()
118 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateAdd()
120 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateAdd()
134 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateMul()
136 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateMul()
|