HomeSort by relevance Sort by last modified time
    Searched refs:mutable_result_layout (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
computation_layout.h 58 ShapeLayout* mutable_result_layout() { return &result_layout_; } function in class:xla::ComputationLayout
layout_assignment_test.cc 85 *computation_layout.mutable_result_layout() = shape_layout;
120 *computation_layout.mutable_result_layout() = col_major;
165 *computation_layout.mutable_result_layout() = shape_layout;
247 TF_CHECK_OK(computation_layout.mutable_result_layout()->CopyLayoutFromShape(
285 TF_CHECK_OK(computation_layout.mutable_result_layout()->CopyLayoutFromShape(
346 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout);
387 *computation_layout.mutable_result_layout() = ShapeLayout(bshape_with_layout);
425 *computation_layout.mutable_result_layout() =
477 *computation_layout.mutable_result_layout() =
628 computation_layout.mutable_result_layout()->ResetLayout
    [all...]
hlo_module.cc 371 TF_RETURN_IF_ERROR(entry_layout->mutable_result_layout()->CopyLayoutFromShape(
service.cc 300 computation_layout->mutable_result_layout()->CopyLayoutFromShape(
303 computation_layout->mutable_result_layout()->Clear();
    [all...]
layout_assignment.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
hlo_test_base.h 188 ->mutable_result_layout()
196 ->mutable_result_layout()
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc 69 *computation_layout.mutable_result_layout() =
156 *computation_layout.mutable_result_layout() = ShapeLayout(result_shape);
223 *computation_layout.mutable_result_layout() =
303 *computation_layout.mutable_result_layout() =
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 75 *computation_layout.mutable_result_layout() =
116 *computation_layout.mutable_result_layout() =
162 *computation_layout.mutable_result_layout() =
194 *computation_layout.mutable_result_layout() =
234 *computation_layout.mutable_result_layout() =
304 *computation_layout.mutable_result_layout() =
  /external/tensorflow/tensorflow/compiler/xla/tools/parser/
hlo_parser.cc 315 ->mutable_result_layout()
    [all...]

Completed in 516 milliseconds