Searched refs:parameter_shape (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dcomputation_layout.h62 const Shape& parameter_shape(int64 param_no) const { function in class:xla::ComputationLayout
H A Dlayout_assignment.cc559 body_layout.parameter_shape(0)));
561 condition_layout.parameter_shape(0)));
567 body_layout.parameter_shape(0))) {
573 ShapeUtil::HumanString(body_layout.parameter_shape(0)).c_str());
582 ShapeUtil::HumanString(condition_layout.parameter_shape(0)).c_str(),
610 true_computation_layout.parameter_shape(0)));
612 false_operand->shape(), false_computation_layout.parameter_shape(0)));
617 true_computation_layout.parameter_shape(0), instruction, 1,
620 false_computation_layout.parameter_shape(0), instruction, 2,
H A Dhlo_module.cc287 const Shape& parameter_shape = local
290 ShapeUtil::Equal(expected_program_shape.parameters(i), parameter_shape))
295 << ", actual: " << ShapeUtil::HumanStringWithLayout(parameter_shape);
H A Dshape_inference.cc1156 const Shape& parameter_shape = to_apply.parameters(i); local
1158 if (!ShapeUtil::IsScalar(parameter_shape)) {
1162 i, ShapeUtil::HumanString(parameter_shape).c_str());
1165 if (!ShapeUtil::SameElementTypeIgnoringFpPrecision(parameter_shape,
1170 i, ShapeUtil::HumanString(parameter_shape).c_str(),
/external/tensorflow/tensorflow/compiler/xla/
H A Dlayout_util.cc136 for (auto& parameter_shape : *program_shape->mutable_parameters()) {
137 LayoutUtil::SetToDefaultLayout(&parameter_shape);
238 for (auto& parameter_shape : *program_shape->mutable_parameters()) {
239 LayoutUtil::ClearLayout(&parameter_shape);
323 for (auto& parameter_shape : program_shape.parameters()) {
324 if (!LayoutUtil::HasLayout(parameter_shape)) {

Completed in 211 milliseconds