HomeSort by relevance Sort by last modified time
    Searched refs:IsNestedTuple (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
infeed_thunk.cc 43 CHECK(!ShapeUtil::IsNestedTuple(hlo_instruction()->shape()));
gpu_transfer_manager.cc 60 if (ShapeUtil::IsNestedTuple(shape)) {
  /external/tensorflow/tensorflow/compiler/xla/
shape_util_test.cc 299 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape({})));
300 EXPECT_FALSE(ShapeUtil::IsNestedTuple(
302 EXPECT_TRUE(ShapeUtil::IsNestedTuple(
304 EXPECT_FALSE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape(
306 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape(
308 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape(
310 EXPECT_TRUE(ShapeUtil::IsNestedTuple(ShapeUtil::MakeTupleShape(
shape_util.h 402 static bool IsNestedTuple(const Shape& shape);
shape_util.cc 320 /* static */ bool ShapeUtil::IsNestedTuple(const Shape& shape) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_transfer_manager.cc 104 if (ShapeUtil::IsNestedTuple(shape)) {
201 if (ShapeUtil::IsNestedTuple(literal_shape)) {
ir_emitter.cc 341 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(shape));
465 TF_RET_CHECK(!ShapeUtil::IsNestedTuple(operand_shape));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
allocation_tracker.cc 111 if (ShapeUtil::IsNestedTuple(shaped_buffer->on_device_shape())) {
hlo_sharding.h 88 CHECK(!ShapeUtil::IsNestedTuple(tuple_shape));

Completed in 380 milliseconds