Home
last modified time | relevance | path

Searched refs:IsNestedTuple (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc339 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape({}))); in TEST()
340 EXPECT_FALSE(ShapeUtil::IsNestedTuple( in TEST()
342 EXPECT_TRUE(ShapeUtil::IsNestedTuple( in TEST()
344 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()
346 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()
348 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()
350 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape( in TEST()
Dshape_util.h524 static bool IsNestedTuple(const Shape& shape);
Dliteral.cc2370 CHECK(!ShapeUtil::IsNestedTuple(*shape_)); in MutableBorrowingLiteral()
2429 CHECK(!ShapeUtil::IsNestedTuple(*shape_)); in BorrowingLiteral()
Dshape_util.cc485 /* static */ bool ShapeUtil::IsNestedTuple(const Shape& shape) { in IsNestedTuple() function in xla::ShapeUtil
/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias.cc35 return shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape); in IsNonNestedTuple()
Dallocation_tracker.cc142 if (ShapeUtil::IsNestedTuple(shaped_buffer->on_device_shape())) { in DeconstructTuple()
Dwhile_loop_simplifier.cc871 if (!ShapeUtil::IsNestedTuple(while_shape)) { in TryFlattenNestedTuples()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_transfer_manager.cc104 if (ShapeUtil::IsNestedTuple(shape)) { in TransferLiteralToInfeed()
200 if (ShapeUtil::IsNestedTuple(literal.shape())) { in TransferLiteralFromOutfeed()
Dir_emitter.cc411 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(data_shape)); in HandleInfeed()
530 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(operand_shape)); in HandleOutfeed()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_execute_op.cc434 TF_RET_CHECK(!xla::ShapeUtil::IsNestedTuple(scoped_buffers.on_host_shape())); in AllocateOutputTensors()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2963 TF_RET_CHECK(shape.IsTuple() && !ShapeUtil::IsNestedTuple(shape)); in AllToAllTuple()