Home
last modified time | relevance | path

Searched defs:shape_with_output_layout (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dclient_library_test_base.cc113 const Shape* shape_with_output_layout) { in ExecuteAndTransfer()
125 const Shape* shape_with_output_layout) { in ExecuteAndTransfer()
133 const Shape* shape_with_output_layout) { in ExecuteAndTransferReference()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_util.cc75 const Shape shape_with_output_layout( in CreateModuleConfig() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient.cc173 std::optional<Shape> shape_with_output_layout; in ExecuteAndTransfer() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dxla.proto455 ShapeProto shape_with_output_layout = 2; field