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

  /external/tensorflow/tensorflow/compiler/xla/
shape_util_test.cc 397 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape( local
402 EXPECT_TRUE(ShapeUtil::Equal(nested_tuple_shape,
403 ShapeUtil::GetSubshape(nested_tuple_shape, {})));
405 array_shape, ShapeUtil::GetSubshape(nested_tuple_shape, {0})));
408 ShapeUtil::GetSubshape(nested_tuple_shape, {1})));
411 ShapeUtil::GetSubshape(nested_tuple_shape, {2, 0})));
426 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape( local
431 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {}));
432 EXPECT_TRUE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {0}));
433 EXPECT_FALSE(ShapeUtil::IsLeafIndex(nested_tuple_shape, {1}))
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_sharding_test.cc 141 // nested_tuple_shape = (f32[], (f32[3]), f32[4, 6])
142 Shape nested_tuple_shape = ShapeUtil::MakeTupleShape({ local
159 tuple_sharding.GetAsShapeTree(nested_tuple_shape);
164 EXPECT_IS_OK(tuple_sharding.Validate(nested_tuple_shape, /*num_devices=*/5));
  /external/tensorflow/tensorflow/compiler/xla/tests/
local_client_execute_test.cc 343 const Shape nested_tuple_shape = local
349 auto param = builder.Parameter(0, nested_tuple_shape, "param");
    [all...]

Completed in 240 milliseconds