Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/
Dshape_layout.h52 Status CopyLayoutFromShape(const Shape& other_shape);
Dshape_layout.cc25 Status ShapeLayout::CopyLayoutFromShape(const Shape& other_shape) { in CopyLayoutFromShape() function in xla::ShapeLayout
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment_test.cc276 TF_CHECK_OK(computation_layout.mutable_result_layout()->CopyLayoutFromShape( in TEST_F()
314 TF_CHECK_OK(computation_layout.mutable_result_layout()->CopyLayoutFromShape( in TEST_F()
730 computation_layout.mutable_parameter_layout(0)->CopyLayoutFromShape( in TEST_F()
867 computation_layout.mutable_parameter_layout(0)->CopyLayoutFromShape( in TEST_F()
911 computation_layout.mutable_parameter_layout(0)->CopyLayoutFromShape( in TEST_F()
Dhlo_module.cc384 entry_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfigFromProto()
387 TF_RETURN_IF_ERROR(entry_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfigFromProto()
Dservice.cc288 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape( in CreateModuleConfig()
298 computation_layout->mutable_result_layout()->CopyLayoutFromShape( in CreateModuleConfig()
Dhlo_parser.cc537 ->CopyLayoutFromShape(param_shape)); in ParseComputations()
542 ->CopyLayoutFromShape(result_shape)); in ParseComputations()