Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dclient_library_test_base.cc114 const Shape* shape_with_output_layout) { in ExecuteAndTransfer()
126 const Shape* shape_with_output_layout) { in ExecuteAndTransfer()
134 const Shape* shape_with_output_layout) { in ExecuteAndTransferReference()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc174 absl::optional<Shape> shape_with_output_layout; in ExecuteAndTransfer() local
/external/tensorflow/tensorflow/compiler/xla/
Dxla.proto284 ShapeProto shape_with_output_layout = 2; field
/external/tensorflow/tensorflow/compiler/xla/service/
Dservice.cc293 const Shape shape_with_output_layout( in CreateModuleConfig() local