| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | symbolic_shapes_test.cc | 32 TensorShapeProto MakeShape(std::vector<int> dims) { in MakeShape() function in tensorflow::grappler::__anonabcc5d180111::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 | 47 Shape matrix = ShapeUtil::MakeShape(F32, {2, 3}); in TEST()
53 auto shape = ShapeUtil::MakeShape(F32, {1, 2, 3, 4}); in TEST()
58 Shape matrix = ShapeUtil::MakeShape(F32, {2, 3}); in TEST()
63 Shape shape = ShapeUtil::MakeShape(F32, {3}); in TEST()
68 Shape shape = ShapeUtil::MakeShape(F32, {3, 2}); in TEST()
74 Shape shape = ShapeUtil::MakeShape(F32, {3, 2, 7}); in TEST()
81 Shape shape = ShapeUtil::MakeShape(F32, {3, 2, 7, 8}); in TEST()
89 Shape shape1 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST()
90 Shape shape2 = ShapeUtil::MakeShape(F32, {3, 2}); in TEST()
98 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()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference_test.cc | 39 const Shape s32_ = ShapeUtil::MakeShape(S32, {});
40 const Shape f16_ = ShapeUtil::MakeShape(F16, {});
41 const Shape f32_ = ShapeUtil::MakeShape(F32, {});
42 const Shape f64_ = ShapeUtil::MakeShape(F64, {});
43 const Shape pred_ = ShapeUtil::MakeShape(PRED, {});
47 const Shape vector_32_ = ShapeUtil::MakeShape(F32, {32});
48 const Shape vector_64_ = ShapeUtil::MakeShape(F32, {64});
49 const Shape matrix_32_48_ = ShapeUtil::MakeShape(F32, {32, 48});
50 const Shape matrix_32_64_ = ShapeUtil::MakeShape(F32, {32, 64});
51 const Shape matrix_64_48_ = ShapeUtil::MakeShape(F32, {64, 48});
[all …]
|
| D | hlo_sharding_util_test.cc | 37 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 3, 5}); in TEST()
38 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 2}); in TEST()
47 Shape input_shape = ShapeUtil::MakeShape(F32, {2, 3, 5}); in TEST()
48 Shape output_shape = ShapeUtil::MakeShape(F32, {3, 5, 2}); in TEST()
56 Shape input_shape = ShapeUtil::MakeShape(F32, {4, 5, 7}); in TEST()
57 Shape output_shape = ShapeUtil::MakeShape(F32, {20, 7}); in TEST()
68 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 7}); in TEST()
69 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 4, 7}); in TEST()
80 Shape input_shape = ShapeUtil::MakeShape(F32, {16, 7}); in TEST()
81 Shape output_shape = ShapeUtil::MakeShape(F32, {4, 4, 7}); in TEST()
[all …]
|
| D | dynamic_dimension_inference_test.cc | 59 0, ShapeUtil::MakeShape(F32, {}), "lhs")); in GetAdd()
61 1, ShapeUtil::MakeShape(F32, {}), "rhs")); in GetAdd()
70 0, ShapeUtil::MakeShape(F32, {}), "lhs")); in GetAddTuple()
73 1, ShapeUtil::MakeShape(F32, {}), "lhs.1")); in GetAddTuple()
75 2, ShapeUtil::MakeShape(F32, {}), "rhs")); in GetAddTuple()
78 3, ShapeUtil::MakeShape(F32, {}), "rhs.1")); in GetAddTuple()
90 0, ShapeUtil::MakeShape(F32, {}), "lhs")); in GetGe()
92 1, ShapeUtil::MakeShape(F32, {}), "rhs")); in GetGe()
94 ShapeUtil::MakeShape(PRED, {}), lhs, rhs, ComparisonDirection::kGe)); in GetGe()
100 const Shape scalar_shape_ = ShapeUtil::MakeShape(S32, {});
[all …]
|
| D | hlo_sharding_test.cc | 66 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4}), in TEST_F()
82 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4}), in TEST_F()
85 sharding.Validate(ShapeUtil::MakeShape(U32, {4}), /*num_devices=*/5)); in TEST_F()
88 sharding.GetAsShapeTree(ShapeUtil::MakeShape(U32, {4})); in TEST_F()
117 EXPECT_IS_NOT_OK(sharding.Validate(ShapeUtil::MakeShape(F32, {4, 6}), in TEST_F()
124 EXPECT_IS_NOT_OK(sharding.Validate(ShapeUtil::MakeShape(U32, {4, 6}), in TEST_F()
130 Shape shape = ShapeUtil::MakeShape(U32, {4, 5}); in TEST_F()
132 EXPECT_IS_OK(sharding.Validate(ShapeUtil::MakeShape(F32, {3, 5}), in TEST_F()
163 ShapeUtil::MakeShape(F32, {}), in TEST_F()
164 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 | hlo_cost_analysis_test.cc | 62 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
73 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
74 auto y = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {}), "y"); in HloCostAnalysisTest()
84 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
95 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
96 auto y = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {}), "y"); in HloCostAnalysisTest()
106 auto x = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {}), "x"); in HloCostAnalysisTest()
107 auto y = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {}), "y"); in HloCostAnalysisTest()
140 auto lhs = Parameter(&builder, 0, ShapeUtil::MakeShape(F32, {10, 5}), "lhs"); in TEST_F()
141 auto rhs = Parameter(&builder, 1, ShapeUtil::MakeShape(F32, {5, 30}), "rhs"); 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 | bfloat16_normalization_test.cc | 96 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
97 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
124 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
125 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
154 Shape f32_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
155 Shape bf16_shape = ShapeUtil::MakeShape(BF16, {2, 4}); in TEST_F()
183 Shape f32_input_shape = ShapeUtil::MakeShape(F32, {2, 4}); in TEST_F()
184 Shape f32_output_shape = ShapeUtil::MakeShape(F32, {4}); in TEST_F()
186 Shape bf16_scalar_shape = ShapeUtil::MakeShape(BF16, {}); in TEST_F()
242 /*parameter_number=*/0, ShapeUtil::MakeShape(F32, {}), "x")); in TEST_F()
[all …]
|
| D | hlo_dce_test.cc | 99 HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
117 HloInstruction::CreateCustomCall(ShapeUtil::MakeShape(F32, {}), in TEST_F()
135 0, ShapeUtil::MakeShape(F32, {}), "live_param")); in TEST_F()
137 1, ShapeUtil::MakeShape(F32, {}), "dead_param1")); in TEST_F()
139 2, ShapeUtil::MakeShape(F32, {}), "dead_param2")); in TEST_F()
215 Shape shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
255 Shape shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
265 HloInstruction::CreateCompare(ShapeUtil::MakeShape(PRED, {}), param, in TEST_F()
313 Shape shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
370 /*parameter_number=*/0, ShapeUtil::MakeShape(F32, {}), "param0")); in TEST_F()
[all …]
|
| D | algebraic_simplifier_test.cc | 60 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F()
365 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F()
521 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F()
543 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F()
565 Shape r0s32 = ShapeUtil::MakeShape(S32, {}); in TEST_F()
593 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
602 Shape r4f32 = ShapeUtil::MakeShape(F32, {4, 5, 6, 7}); in TEST_F()
606 Shape r3f32 = ShapeUtil::MakeShape(F32, {5, 6, 7}); in TEST_F()
610 Shape r1f32 = ShapeUtil::MakeShape(F32, {5}); in TEST_F()
624 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); 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 | 46 auto shape = ShapeUtil::MakeShape(F32, {1}); in TEST_F()
97 ShapeUtil::MakeShape(F32, {1}), {c1.get(), c2.get()}, "foo_target"); in TEST_F()
126 EXPECT_THAT(p0.get(), op::Shape(ShapeUtil::MakeShape(F32, {5, 7}))); in TEST_F()
130 ::testing::Not(op::ShapeWithLayout(ShapeUtil::MakeShape(F32, {5, 7})))); in TEST_F()
133 ::testing::Not(op::Shape(ShapeUtil::MakeShape(F32, {7, 5})))); in TEST_F()
137 ::testing::Not(op::ShapeWithLayout(ShapeUtil::MakeShape(F32, {7, 5})))); in TEST_F()
150 EXPECT_THAT(Explain(p0.get(), op::Shape(ShapeUtil::MakeShape(F32, {7, 5}))), in TEST_F()
161 auto p0 = HloInstruction::CreateParameter(0, ShapeUtil::MakeShape(F32, {5}), in TEST_F()
164 auto p1 = HloInstruction::CreateParameter(1, ShapeUtil::MakeShape(F32, {7}), in TEST_F()
169 {ShapeUtil::MakeShape(F32, {7}), ShapeUtil::MakeShape(S32, {9}), in TEST_F()
[all …]
|
| D | hlo_instruction_test.cc | 47 Shape r0f32_ = ShapeUtil::MakeShape(F32, {});
331 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F()
332 Shape f32a100x10 = ShapeUtil::MakeShape(F32, {100, 10}); in TEST_F()
369 Shape r0f32 = ShapeUtil::MakeShape(F32, {}); in TEST_F()
370 Shape f32v100 = ShapeUtil::MakeShape(F32, {100}); in TEST_F()
371 Shape f32a100x10 = ShapeUtil::MakeShape(F32, {100, 10}); in TEST_F()
765 auto clone_shape = ShapeUtil::MakeShape(F32, {3, 3}); in TEST_F()
784 auto clone_shape = ShapeUtil::MakeShape(F32, {2, 2}); in TEST_F()
804 auto clone_shape = ShapeUtil::MakeShape(F32, {1, 2, 3}); in TEST_F()
812 const Shape scalar_shape = ShapeUtil::MakeShape(F32, {}); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder_test.cc | 142 Shape s = ShapeUtil::MakeShape(PRED, {}); in TEST_F()
158 Shape s = ShapeUtil::MakeShape(PRED, {}); in TEST_F()
184 auto x = Parameter(&b, 0, ShapeUtil::MakeShape(F32, {3, 5}), "x"); in TEST_F()
193 const auto& x_shape = ShapeUtil::MakeShape(S32, {2, 4, 6}); in TEST_F()
194 const auto& y_shape = ShapeUtil::MakeShape(S32, {2, 4}); in TEST_F()
209 auto x = Parameter(&b, 0, ShapeUtil::MakeShape(S32, {1, 3, 5, 7}), "x"); in TEST_F()
218 auto x = Parameter(&b, 0, ShapeUtil::MakeShape(U32, {2, 4, 6}), "x"); in TEST_F()
219 auto y = Parameter(&b, 1, ShapeUtil::MakeShape(U32, {2, 4}), "y"); in TEST_F()
228 auto x = Parameter(&b, 0, ShapeUtil::MakeShape(F32, {1}), "x"); in TEST_F()
229 auto y = Parameter(&b, 1, ShapeUtil::MakeShape(S32, {}), "dyn_dim"); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_instruction_fusion_test.cc | 53 0, ShapeUtil::MakeShape(F32, {1024, 256}), "arg0")); in TEST_F()
55 1, ShapeUtil::MakeShape(F32, {256}), "arg1")); in TEST_F()
58 ShapeUtil::MakeShape(F32, {1024, 256}), HloOpcode::kExp, arg0)); in TEST_F()
60 MakeDot(ShapeUtil::MakeShape(F32, {1024}), exp0, arg1)); in TEST_F()
72 0, ShapeUtil::MakeShape(F32, {256}), "arg0")); in TEST_F()
74 1, ShapeUtil::MakeShape(F32, {256, 1024}), "arg1")); in TEST_F()
77 ShapeUtil::MakeShape(F32, {256, 1024}), HloOpcode::kExp, arg1)); in TEST_F()
79 MakeDot(ShapeUtil::MakeShape(F32, {1024}), arg0, exp1)); in TEST_F()
91 0, ShapeUtil::MakeShape(F32, {2, 512, 2, 128}), "arg0")); in TEST_F()
93 1, ShapeUtil::MakeShape(F32, {256}), "arg1")); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_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()
144 0, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "activations")); in TEST_F()
147 1, ShapeUtil::MakeShape(F32, {1, 1, 3, 1}), "gradients")); in TEST_F()
175 0, ShapeUtil::MakeShape(F32, {20, 35, 35, 32}), "activations")); in TEST_F()
178 1, ShapeUtil::MakeShape(F32, {20, 35, 35, 32}), "gradients")); in TEST_F()
187 ShapeUtil::MakeShape(F32, {32, 3, 3, 32}), activations, gradients, in TEST_F()
205 0, ShapeUtil::MakeShape(F32, {20, 10, 10, 192}), "activations")); in TEST_F()
208 1, ShapeUtil::MakeShape(F32, {20, 4, 4, 320}), "gradients")); in TEST_F()
217 ShapeUtil::MakeShape(F32, {320, 3, 3, 192}), activations, gradients, in TEST_F()
[all …]
|
| D | custom_call_test.cc | 65 ShapeUtil::MakeShape(F32, {}), in TEST_F()
75 ShapeUtil::MakeShape(F32, {}), in TEST_F()
92 ShapeUtil::MakeShape(F32, {128}), /*opaque=*/""); in TEST_F()
108 ShapeUtil::MakeShape(F32, {}), kExpectedOpaque); in TEST_F()
154 ShapeUtil::MakeShape(F32, {8}), in TEST_F()
156 ShapeUtil::MakeShape(F32, {128}), in TEST_F()
157 ShapeUtil::MakeShape(F32, {256}), in TEST_F()
159 ShapeUtil::MakeShape(F32, {1024}), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | cpu_gpu_fusion_test.cc | 101 ShapeUtil::MakeShape(prim_type, {test_width, test_height}); in TestElementwise2D()
207 ShapeUtil::MakeShape(F32, {3, 1}), HloOpcode::kAdd, const0, const1)); in XLA_TEST_F()
209 ShapeUtil::MakeShape(F32, {1, 3}), add2, {1, 0})); in XLA_TEST_F()
213 ShapeUtil::MakeShape(F32, {2, 3}), {reshape3, const4}, 0)); in XLA_TEST_F()
217 ShapeUtil::MakeShape(F32, {2, 3}), HloOpcode::kNegate, const6)); in XLA_TEST_F()
219 ShapeUtil::MakeShape(F32, {2, 3}), HloOpcode::kAdd, concat5, negate7)); in XLA_TEST_F()
226 HloInstruction::CreateTernary(ShapeUtil::MakeShape(F32, {2, 3}), in XLA_TEST_F()
229 ShapeUtil::MakeShape(F32, {2, 1}), select11, {0, 1}, {2, 2}, {1, 1})); in XLA_TEST_F()
253 ShapeUtil::MakeShape(F32, {1, 3}), HloOpcode::kCopy, const0)); in XLA_TEST_F()
258 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 | 93 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateMax()
95 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateMax()
109 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateMin()
111 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateMin()
125 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateAdd()
127 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateAdd()
141 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateMul()
143 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateMul()
|