Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc325 Shape layout_shape; in ComputeAndCompareLiteralWithStatus() local
330 layout_shape = *shape_with_layout; in ComputeAndCompareLiteralWithStatus()
332 &layout_shape, [&](Shape* subshape, const ShapeIndex& /*index*/) { in ComputeAndCompareLiteralWithStatus()
337 shape_with_layout = &layout_shape; in ComputeAndCompareLiteralWithStatus()
382 Shape layout_shape; in ComputeAndCompareLiteralWithStatus() local
387 layout_shape = *shape_with_layout; in ComputeAndCompareLiteralWithStatus()
389 &layout_shape, [&](Shape* subshape, const ShapeIndex& /*index*/) { in ComputeAndCompareLiteralWithStatus()
394 shape_with_layout = &layout_shape; in ComputeAndCompareLiteralWithStatus()