Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_alias_analysis_test.cc255 const Shape tuple_shape = in TEST_F() local
260 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
304 const Shape tuple_shape = in TEST_F() local
309 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
362 const Shape tuple_shape = in TEST_F() local
368 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_F()
382 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_F()
390 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
393 HloInstruction::CreateWhile(tuple_shape, condition, body, param)); in TEST_F()
547 const Shape tuple_shape = in TEST_F() local
[all …]
Dtuple_util.cc64 const Shape& tuple_shape = tuple->shape(); in ReplaceTupleWith() local
91 std::vector<HloInstruction*> tuple_args(tuple_shape.tuple_shapes_size()); in ReplaceTupleWith()
92 CHECK_GE(tuple_shape.tuple_shapes_size(), shape_index[0]); in ReplaceTupleWith()
93 for (int i = 0; i < tuple_shape.tuple_shapes_size(); ++i) { in ReplaceTupleWith()
94 const Shape& subshape = tuple_shape.tuple_shapes(i); in ReplaceTupleWith()
138 if (shape_index[0] == tuple_shape.tuple_shapes_size()) { in ReplaceTupleWith()
Dpattern_matcher_test.cc105 auto tuple_shape = ShapeUtil::MakeTupleShape({ in TEST_F() local
109 EXPECT_TRUE(Match(&tuple_shape, match::Shape().IsTuple())); in TEST_F()
110 EXPECT_FALSE(Match(&tuple_shape, match::Shape().IsArray())); in TEST_F()
111 EXPECT_FALSE(Match(&tuple_shape, match::Shape().IsScalar())); in TEST_F()
115 &tuple_shape, in TEST_F()
120 ShapeUtil::Equal(*subshape, ShapeUtil::GetSubshape(tuple_shape, {0}))); in TEST_F()
121 EXPECT_TRUE(Match(&tuple_shape, in TEST_F()
124 &ShapeUtil::GetSubshape(tuple_shape, {0}))))); in TEST_F()
125 EXPECT_FALSE(Match(&tuple_shape, in TEST_F()
128 &ShapeUtil::GetSubshape(tuple_shape, {1}))))); in TEST_F()
[all …]
Dmemory_space_assignment_test.cc777 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, scalar_shape}); in TEST_P() local
782 HloInstruction::CreateParameter(0, tuple_shape, "cond_param")); in TEST_P()
797 HloInstruction::CreateParameter(0, tuple_shape, "body_param")); in TEST_P()
832 tuple_shape, cond_computation, body_computation, tuple)); in TEST_P()
854 EXPECT_THAT(tuple, op::ShapeWithLayout(tuple_shape)); in TEST_P()
872 Shape tuple_shape = in TEST_P() local
875 HloInstruction::CreateParameter(0, tuple_shape, "p")); in TEST_P()
1062 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
1067 HloInstruction::CreateParameter(0, tuple_shape, "p")); in TEST_P()
1213 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape, shape}); in TEST_P() local
[all …]
Dhlo_sharding.cc257 HloSharding HloSharding::Tuple(const Shape& tuple_shape, in Tuple() argument
259 CHECK(tuple_shape.IsTuple()) << ShapeUtil::HumanString(tuple_shape); in Tuple()
262 << sharding.ToString() << ShapeUtil::HumanString(tuple_shape); in Tuple()
265 CHECK_EQ(flattened_list.size(), RequiredLeaves(tuple_shape)) in Tuple()
267 << RequiredLeaves(tuple_shape); in Tuple()
271 HloSharding HloSharding::SingleTuple(const Shape& tuple_shape, in SingleTuple() argument
273 CHECK(tuple_shape.IsTuple()) << ShapeUtil::HumanString(tuple_shape); in SingleTuple()
275 int64_t leaf_count = RequiredLeaves(tuple_shape); in SingleTuple()
Dhlo_dataflow_analysis_test.cc396 const Shape tuple_shape = in TEST_P() local
402 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_P()
416 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_P()
430 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
499 const Shape tuple_shape = in TEST_P() local
505 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_P()
518 HloInstruction::CreateParameter(0, tuple_shape, "param")); in TEST_P()
532 HloInstruction::CreateWhile(tuple_shape, condition, body, tuple)); in TEST_P()
534 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while0)); in TEST_P()
536 HloInstruction::CreateWhile(tuple_shape, condition, body, xla_while1)); in TEST_P()
[all …]
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.cc130 Shape tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
134 HloInstruction::CreateConstant(Literal::CreateFromShape(tuple_shape))); in TEST_F()
Dshaped_buffer_test.cc159 Shape tuple_shape = in TEST() local
162 ScopedShapedBuffer sb(tuple_shape, &allocator, /*device_ordinal=*/0); in TEST()
Dhlo_matchers_test.cc169 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_F() local
172 auto p2 = HloInstruction::CreateParameter(1, tuple_shape, "param.2"); in TEST_F()
176 tuple_shape, {HloSharding::Tile(assignment), HloSharding::AssignDevice(1), in TEST_F()
Dhlo_sharding_test.cc413 auto tuple_shape = in TEST_P() local
418 tuple_shape, in TEST_P()
424 auto tuple_shape = ShapeUtil::MakeTupleShape( in TEST_P() local
431 ShapeTree<HloSharding> sharding_tree(tuple_shape, HloSharding::Replicate()); in TEST_P()
Dhlo_sharding.h96 static HloSharding Tuple(const Shape& tuple_shape,
101 static HloSharding SingleTuple(const Shape& tuple_shape,
Dcopy_insertion_test.cc1422 const Shape tuple_shape = in TEST_F() local
1427 HloInstruction::CreateParameter(0, tuple_shape, "0")); in TEST_F()
1455 const Shape tuple_shape = in TEST_F() local
1460 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
1488 const Shape tuple_shape = in TEST_F() local
1493 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
1521 const Shape tuple_shape = in TEST_F() local
1526 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
1556 const Shape tuple_shape = in TEST_F() local
1561 HloInstruction::CreateParameter(0, tuple_shape, "p0")); in TEST_F()
[all …]
Dheap_simulator_test.cc44 const Shape tuple_shape = in TEST_F() local
50 HloInstruction::CreateParameter(0, tuple_shape, "cond_param")); in TEST_F()
65 HloInstruction::CreateParameter(0, tuple_shape, "body_param")); in TEST_F()
81 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()
Dtriangular_solve_expander.cc302 Shape tuple_shape = ShapeUtil::MakeTupleShape(tuple_shapes); in InvertDiagonalBlocks() local
312 Parameter(condb.get(), 0, tuple_shape, "InvertDiagCondTuple"), 0); in InvertDiagonalBlocks()
322 Parameter(bodyb.get(), 0, tuple_shape, "InvertDiagBodyTuple"); in InvertDiagonalBlocks()
/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/compiler/xla/tests/
Dconditional_test.cc82 const Shape& tuple_shape) { in CreateTupleCeilComputation() argument
84 auto tuple = Parameter(&builder, 0, tuple_shape, "tuple"); in CreateTupleCeilComputation()
104 const std::string& computation_name, const Shape& tuple_shape) { in CreateTupleFloorComputation() argument
106 auto tuple = Parameter(&builder, 0, tuple_shape, "tuple"); in CreateTupleFloorComputation()
126 const Shape& tuple_shape) { in CreateTupleAddComputation() argument
128 auto tuple = Parameter(&builder, 0, tuple_shape, "tuple"); in CreateTupleAddComputation()
146 const Shape& tuple_shape) { in CreateTupleSubComputation() argument
148 auto tuple = Parameter(&builder, 0, tuple_shape, "tuple"); in CreateTupleSubComputation()
636 Shape tuple_shape = ShapeUtil::MakeTupleShape({r0bool, r0f32_, r0f32_}); in XLA_TEST_F() local
637 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()
460 const Shape tuple_shape = ShapeUtil::MakeTupleShape(inner_tuple_shapes); in XLA_TEST_F() local
463 auto param = Parameter(&builder, 0, tuple_shape, "param"); in XLA_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/service/llvm_ir/
Dtuple_ops.cc63 const Shape& tuple_shape, llvm::IRBuilder<>* b) { in EmitTupleAllocasAtFunctionEntry() argument
70 CHECK(tuple_shape.IsTuple()); in EmitTupleAllocasAtFunctionEntry()
71 int tuple_size = tuple_shape.tuple_shapes_size(); in EmitTupleAllocasAtFunctionEntry()
75 const Shape& element_shape = tuple_shape.tuple_shapes(i); in EmitTupleAllocasAtFunctionEntry()
Dtuple_ops.h38 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.cc461 Shape tuple_shape = in TEST() local
464 ShapeUtil::Equal(tuple_shape, ShapeUtil::GetSubshape(tuple_shape, {}))); in TEST()
466 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {0}))); in TEST()
468 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {1}))); in TEST()
470 ShapeUtil::Equal(array_shape, ShapeUtil::GetSubshape(tuple_shape, {2}))); in TEST()
496 Shape tuple_shape = ShapeUtil::MakeTupleShape({array_shape, array_shape}); in TEST() local
497 EXPECT_FALSE(ShapeUtil::IsLeafIndex(tuple_shape, {})); in TEST()
498 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {0})); in TEST()
499 EXPECT_TRUE(ShapeUtil::IsLeafIndex(tuple_shape, {1})); in TEST()
751 Shape tuple_shape = ShapeUtil::MakeTupleShape({shape}); in TEST() local
[all …]
/external/tensorflow/tensorflow/compiler/xla/tools/
Ddriver.cc489 TupleShape tuple_shape = assignment.buffers_shape[allocation_idx]; in main() local
490 Check(tuple_shape.elements.size() == 1, in main()
492 TupleShapeToString(tuple_shape)); in main()
493 ArrayShape shape = tuple_shape.elements[0]; in main()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client.cc185 std::vector<BorrowingLiteral> leaves, const Shape& tuple_shape, in FromLiterals() argument
189 VLOG(1) << "PyTpuBuffer::FromLiterals: shape: " << tuple_shape.DebugString() in FromLiterals()
195 if (!tuple_shape.IsTuple()) { in FromLiterals()
198 tuple_shape, in FromLiterals()
199 [driver, &leaves, &tuple_shape, in FromLiterals()
216 ShapeUtil::GetLeafShapes(tuple_shape)) { in FromLiterals()
264 Shape tuple_shape = ShapeUtil::MakeTupleShape(child_shapes); in MakeTuple() local
272 tuple_shape, std::move(tuple_device_buffer), in MakeTuple()

12