Home
last modified time | relevance | path

Searched refs:tuple_shape (Results 1 – 25 of 45) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc222 const Shape tuple_shape = in TEST_F() local
227 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
271 const Shape tuple_shape = in TEST_F() local
276 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
329 const Shape tuple_shape = in TEST_F() local
335 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_F()
349 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_F()
357 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
360 HloInstruction::CreateWhile(tuple_shape, condition, body, param)); in TEST_F()
515 const Shape tuple_shape = in TEST_F() local
[all …]
Dpattern_matcher_test.cc106 auto tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F() local
110 EXPECT_TRUE(Match(&tuple_shape, match::Shape().IsTuple())); in TEST_F()
111 EXPECT_FALSE(Match(&tuple_shape, match::Shape().IsArray())); in TEST_F()
112 EXPECT_FALSE(Match(&tuple_shape, match::Shape().IsScalar())); in TEST_F()
116 &tuple_shape, in TEST_F()
121 ShapeUtil::Equal(*subshape, ShapeUtil::GetSubshape(tuple_shape, {0}))); in TEST_F()
122 EXPECT_TRUE(Match(&tuple_shape, in TEST_F()
125 &ShapeUtil::GetSubshape(tuple_shape, {0}))))); in TEST_F()
126 EXPECT_FALSE(Match(&tuple_shape, in TEST_F()
129 &ShapeUtil::GetSubshape(tuple_shape, {1}))))); in TEST_F()
[all …]
Dhlo_dataflow_analysis_test.cc389 const Shape tuple_shape = in TEST_P() local
395 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_P()
409 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_P()
423 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
492 const Shape tuple_shape = in TEST_P() local
498 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_P()
511 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_P()
525 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
527 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0)); in TEST_P()
529 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1)); in TEST_P()
[all …]
Dmemory_space_assignment_test.cc761 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_P() local
766 HloInstruction::CreateParameter(0, tuple_shape, "cond_param")); in TEST_P()
781 HloInstruction::CreateParameter(0, tuple_shape, "body_param")); in TEST_P()
816 tuple_shape, cond_computation, body_computation, tuple)); in TEST_P()
838 EXPECT_THAT(tuple, op::ShapeWithLayout(tuple_shape)); in TEST_P()
856 Shape tuple_shape = in TEST_P() local
859 HloInstruction::CreateParameter(0, tuple_shape, "p")); in TEST_P()
1046 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
1051 HloInstruction::CreateParameter(0, tuple_shape, "p")); in TEST_P()
1197 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
[all …]
Dhlo_sharding.cc169 HloSharding HloSharding::Tuple(const Shape& tuple_shape, in Tuple() argument
171 CHECK(tuple_shape.IsTuple()) << ShapeUtil::HumanString(tuple_shape); in Tuple()
174 << sharding.ToString() << ShapeUtil::HumanString(tuple_shape); in Tuple()
177 CHECK_EQ(flattened_list.size(), RequiredLeaves(tuple_shape)) in Tuple()
179 << RequiredLeaves(tuple_shape); in Tuple()
183 HloSharding HloSharding::SingleTuple(const Shape& tuple_shape, in SingleTuple() argument
185 CHECK(tuple_shape.IsTuple()) << ShapeUtil::HumanString(tuple_shape); in SingleTuple()
187 int64_t leaf_count = RequiredLeaves(tuple_shape); in SingleTuple()
Dhlo_live_range_test.cc267 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_F() local
271 HloInstruction::CreateParameter(0, tuple_shape, "cond_param")); in TEST_F()
284 HloInstruction::CreateParameter(0, tuple_shape, "body_param")); in TEST_F()
319 tuple_shape, cond_computation, body_computation, tuple)); in TEST_F()
Dhlo_graph_dumper_test.cc129 Shape tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
133 HloInstruction::CreateConstant(Literal::CreateFromShape(tuple_shape))); in TEST_F()
Dshaped_buffer_test.cc156 Shape tuple_shape = in TEST() local
159 ScopedShapedBuffer sb(tuple_shape, &allocator, /*device_ordinal=*/0); in TEST()
Dhlo_matchers_test.cc168 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
171 auto p2 = HloInstruction::CreateParameter(1, tuple_shape, "param.2"); in TEST_F()
175 tuple_shape, {HloSharding::Tile(assignment), HloSharding::AssignDevice(1), in TEST_F()
Dtuple_points_to_analysis_test.cc440 Shape tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
445 HloInstruction::CreateParameter(0, tuple_shape, "param0")); in TEST_F()
447 HloInstruction::CreateParameter(1, tuple_shape, "param1")); in TEST_F()
451 tuple_shape, HloOpcode::kTupleSelect, pred, param0, param1)); in TEST_F()
453 HloInstruction::CreateUnary(tuple_shape, HloOpcode::kCopy, select)); in TEST_F()
679 Shape tuple_shape = in Run() local
685 HloInstruction::CreateParameter(0, tuple_shape, "param0")); in Run()
Dhlo_sharding_test.cc406 auto tuple_shape = in TEST_P() local
411 tuple_shape, in TEST_P()
417 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_P() local
424 ShapeTree<HloSharding> sharding_tree(tuple_shape, HloSharding::Replicate()); in TEST_P()
Dhlo_sharding.h100 static HloSharding Tuple(const Shape& tuple_shape,
105 static HloSharding SingleTuple(const Shape& tuple_shape,
Dheap_simulator_test.cc45 const Shape tuple_shape = in TEST_F() local
51 HloInstruction::CreateParameter(0, tuple_shape, "cond_param")); in TEST_F()
66 HloInstruction::CreateParameter(0, tuple_shape, "body_param")); in TEST_F()
82 tuple_shape, cond_computation, body_computation, tuple)); in TEST_F()
889 const Shape tuple_shape = in TEST_F() local
894 HloInstruction::CreateParameter(0, tuple_shape, "cond_param")); in TEST_F()
907 HloInstruction::CreateParameter(0, tuple_shape, "body_param")); in TEST_F()
913 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_F()
915 tuple_shape, cond_computation, body_computation, param)); in TEST_F()
Dcopy_insertion_test.cc1576 const Shape tuple_shape = in TEST_F() local
1581 HloInstruction::CreateParameter(0, tuple_shape, "0")); in TEST_F()
1609 const Shape tuple_shape = in TEST_F() local
1614 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
1642 const Shape tuple_shape = in TEST_F() local
1647 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
1675 const Shape tuple_shape = in TEST_F() local
1680 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
1710 const Shape tuple_shape = in TEST_F() local
1715 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
[all …]
Dbuffer_assignment_test.cc1598 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F() local
1611 HloInstruction::CreateCall(tuple_shape, {param}, sub_computation)); in TEST_F()
1639 auto tuple_shape = ShapeUtil::MakeTupleShape({elem_shape}); in TEST_F() local
1643 HloInstruction::CreateParameter(0, tuple_shape, "d_param")); in TEST_F()
1648 HloInstruction::CreateParameter(0, tuple_shape, "c_param")); in TEST_F()
1650 HloInstruction::CreateCall(tuple_shape, {c_param}, d_computation.get())); in TEST_F()
1655 HloInstruction::CreateParameter(0, tuple_shape, "b_param")); in TEST_F()
1657 HloInstruction::CreateCall(tuple_shape, {b_param}, c_computation.get())); in TEST_F()
1666 HloInstruction::CreateCall(tuple_shape, {a_tuple}, b_computation.get())); in TEST_F()
1724 auto tuple_shape = in TEST_F() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_bytesizeof_test.cc25 auto tuple_shape = in TEST_F() local
27 EXPECT_EQ(xla::llvm_ir::ByteSizeOf(tuple_shape, data_layout), in TEST_F()
33 auto tuple_shape = in TEST_F() local
35 EXPECT_EQ(xla::llvm_ir::ByteSizeOf(tuple_shape, data_layout), in TEST_F()
/external/tensorflow/tensorflow/core/tpu/kernels/xla/
Doutfeed_ops.cc72 xla::Shape tuple_shape = xla::ShapeUtil::MakeTupleShape(xla_shapes); in Compile() local
74 << xla::ShapeUtil::HumanStringWithLayout(tuple_shape); in Compile()
79 xla::Outfeed(tuple, tuple_shape, outfeed_config); in Compile()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconditional_test.cc80 const Shape& tuple_shape) { in CreateTupleCeilComputation() argument
82 auto tuple = Parameter(&builder, 0, tuple_shape, "tuple"); in CreateTupleCeilComputation()
102 const Shape& tuple_shape) { in CreateTupleFloorComputation() argument
104 auto tuple = Parameter(&builder, 0, tuple_shape, "tuple"); in CreateTupleFloorComputation()
124 const Shape& tuple_shape) { in CreateTupleAddComputation() argument
126 auto tuple = Parameter(&builder, 0, tuple_shape, "tuple"); in CreateTupleAddComputation()
144 const Shape& tuple_shape) { in CreateTupleSubComputation() argument
146 auto tuple = Parameter(&builder, 0, tuple_shape, "tuple"); in CreateTupleSubComputation()
635 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F() local
636 auto param0 = Parameter(&inner_builder, 0, tuple_shape, "param0"); in XLA_TEST_F()
[all …]
Dlocal_client_execute_test.cc372 const Shape tuple_shape = in XLA_TEST_F() local
376 auto param = Parameter(&builder, 0, tuple_shape, "param"); in XLA_TEST_F()
416 const Shape tuple_shape = ShapeUtil::MakeTupleShape(element_shapes); in XLA_TEST_F() local
419 auto param = Parameter(&builder, 0, tuple_shape, "param"); in XLA_TEST_F()
458 const Shape tuple_shape = ShapeUtil::MakeTupleShape(inner_tuple_shapes); in XLA_TEST_F() local
461 auto param = Parameter(&builder, 0, tuple_shape, "param"); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dtuple_ops.cc85 const Shape& tuple_shape, llvm::IRBuilder<>* b) { in EmitTupleAllocasAtFunctionEntry() argument
92 CHECK(tuple_shape.IsTuple()); in EmitTupleAllocasAtFunctionEntry()
93 int tuple_size = tuple_shape.tuple_shapes_size(); in EmitTupleAllocasAtFunctionEntry()
97 const Shape& element_shape = tuple_shape.tuple_shapes(i); in EmitTupleAllocasAtFunctionEntry()
Dtuple_ops.h75 const Shape& tuple_shape, llvm::IRBuilder<>* b);
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dloops.cc37 Shape tuple_shape = ShapeUtil::MakeTupleShape(var_shapes); in WhileLoopHelper() local
52 auto parameter = Parameter(cond_builder.get(), 0, tuple_shape, "parameter"); in WhileLoopHelper()
65 auto parameter = Parameter(body_builder.get(), 0, tuple_shape, "parameter"); in WhileLoopHelper()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc446 Shape tuple_shape = in TEST() local
449 ShapeUtil::Equal(tuple_shape, ShapeUtil::GetSubshape(tuple_shape, {}))); in TEST()
451 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {0}))); in TEST()
453 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {1}))); in TEST()
455 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {2}))); in TEST()
481 Shape tuple_shape = ShapeUtil::MakeTupleShape({array_shape, array_shape}); in TEST() local
482 EXPECT_FALSE(ShapeUtil::IsLeafIndex(tuple_shape, {})); in TEST()
483 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {0})); in TEST()
484 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {1})); in TEST()
737 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape}); in TEST() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc175 std::vector<BorrowingLiteral> leaves, const Shape& tuple_shape, in FromLiterals() argument
179 VLOG(1) << "PyTpuBuffer::FromLiterals: shape: " << tuple_shape.DebugString() in FromLiterals()
185 if (!tuple_shape.IsTuple()) { in FromLiterals()
188 tuple_shape, in FromLiterals()
189 [driver, &leaves, &tuple_shape, in FromLiterals()
206 ShapeUtil::GetLeafShapes(tuple_shape)) { in FromLiterals()
254 Shape tuple_shape = ShapeUtil::MakeTupleShape(child_shapes); in MakeTuple() local
262 tuple_shape, std::move(tuple_device_buffer), in MakeTuple()
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc470 TupleShape tuple_shape = assignment.buffers_shape[allocation_idx]; in main() local
471 Check(tuple_shape.elements.size() == 1, in main()
473 TupleShapeToString(tuple_shape)); in main()
474 ArrayShape shape = tuple_shape.elements[0]; in main()

12