HomeSort by relevance Sort by last modified time
    Searched full:mutable_parameter_layout (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_layout_assignment_test.cc 73 *computation_layout.mutable_parameter_layout(0) =
114 *computation_layout.mutable_parameter_layout(0) =
158 *computation_layout.mutable_parameter_layout(0) =
160 *computation_layout.mutable_parameter_layout(1) =
192 *computation_layout.mutable_parameter_layout(0) =
232 *computation_layout.mutable_parameter_layout(0) =
300 *computation_layout.mutable_parameter_layout(0) =
302 *computation_layout.mutable_parameter_layout(1) =
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_layout_assignment_test.cc 65 *computation_layout.mutable_parameter_layout(0) =
67 *computation_layout.mutable_parameter_layout(1) =
151 *computation_layout.mutable_parameter_layout(0) =
218 *computation_layout.mutable_parameter_layout(0) =
298 *computation_layout.mutable_parameter_layout(constrained_param_no) =
  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 83 *computation_layout.mutable_parameter_layout(0) = shape_layout;
84 *computation_layout.mutable_parameter_layout(1) = shape_layout;
118 *computation_layout.mutable_parameter_layout(0) = col_major;
119 *computation_layout.mutable_parameter_layout(1) = row_major;
344 *computation_layout.mutable_parameter_layout(0) =
385 *computation_layout.mutable_parameter_layout(0) =
423 *computation_layout.mutable_parameter_layout(0) =
475 *computation_layout.mutable_parameter_layout(0) =
541 *computation_layout.mutable_parameter_layout(0) =
543 *computation_layout.mutable_parameter_layout(1)
    [all...]
computation_layout.h 44 ShapeLayout* mutable_parameter_layout(int64 param_no) { function in class:xla::ComputationLayout
hlo_module.cc 368 entry_layout->mutable_parameter_layout(i)->CopyLayoutFromShape(
service.cc 290 computation_layout->mutable_parameter_layout(i)->CopyLayoutFromShape(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
hlo_test_base.h 180 ->mutable_parameter_layout(param_no)
  /external/tensorflow/tensorflow/compiler/xla/tools/parser/
hlo_parser.cc 308 ->mutable_parameter_layout(p)
    [all...]

Completed in 438 milliseconds