Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dcomputation_layout.h46 void add_parameter_layout(ShapeLayout shape_layout) { in add_parameter_layout() function
Dhlo_computation.cc171 config.mutable_entry_computation_layout()->add_parameter_layout( in AddEntryComputationParameter()
Dhlo_parser.cc794 computation_layout->add_parameter_layout(ShapeLayout(param)); in ParseComputationLayout()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc460 computation_layout.add_parameter_layout(xla::ShapeLayout(shape)); in TPUExecute()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dc_api_conversions.cc447 layout.add_parameter_layout(xla::ShapeLayout( in ConfigWithLayout()