Home
last modified time | relevance | path

Searched refs:add_parameter_layout (Results 1 – 4 of 4) 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.cc137 config.mutable_entry_computation_layout()->add_parameter_layout( in AddEntryComputationParameter()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc456 computation_layout.add_parameter_layout(xla::ShapeLayout(shape)); in TPUExecute()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc407 layout.add_parameter_layout(xla::ShapeLayout( in ConfigWithLayout()