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

  /external/tensorflow/tensorflow/compiler/xla/service/
computation_layout.cc 31 SetToDefaultLayout();
34 void ComputationLayout::SetToDefaultLayout() {
36 parameter_layout.SetToDefaultLayout();
38 result_layout_.SetToDefaultLayout();
computation_layout.h 68 void SetToDefaultLayout();
local_service.cc 132 LayoutUtil::SetToDefaultLayout(
layout_assignment.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_layout.h 59 void SetToDefaultLayout();
shape_layout.cc 45 void ShapeLayout::SetToDefaultLayout() {
46 LayoutUtil::SetToDefaultLayout(&shape_);
layout_util.h 53 static void SetToDefaultLayout(Shape* shape);
61 static void SetToDefaultLayout(ProgramShape* program_shape);
layout_util.cc 41 // Internal helper for GetDefaultLayoutForShape and SetToDefaultLayout. Sets
111 /* static */ void LayoutUtil::SetToDefaultLayout(Shape* shape) {
115 SetToDefaultLayout(&element_shape);
131 LayoutUtil::SetToDefaultLayout(&copy);
135 /* static */ void LayoutUtil::SetToDefaultLayout(ProgramShape* program_shape) {
137 LayoutUtil::SetToDefaultLayout(&parameter_shape);
139 LayoutUtil::SetToDefaultLayout(program_shape->mutable_result());
layout_util_test.cc 248 LayoutUtil::SetToDefaultLayout(&shape);
262 LayoutUtil::SetToDefaultLayout(&shape_with_layout);
literal_util_test.cc     [all...]
shape_util.cc 234 LayoutUtil::SetToDefaultLayout(shape);
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_compiler.cc 660 xla::LayoutUtil::SetToDefaultLayout(&result->xla_output_shape);
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_unnested.cc 547 LayoutUtil::SetToDefaultLayout(input_shape);
    [all...]

Completed in 895 milliseconds