Searched refs:result_layout (Results 1 – 13 of 13) sorted by relevance
57 const Shape* ExecutableBuildOptions::result_layout() const { in result_layout() function in xla::ExecutableBuildOptions68 string result_layout = "nullopt"; in ToString() local70 result_layout = ShapeUtil::HumanStringWithLayout(result_layout_); in ToString()75 device_ordinal_, result_layout, num_replicas_); in ToString()
38 service_instance.result_layout = instance.result_layout; in CompileAheadOfTime()
47 const Shape* result_layout() const;
46 const Shape* result_layout; member
36 explicit ComputationLayout(ShapeLayout result_layout) in ComputationLayout() argument37 : result_layout_(std::move(result_layout)) {} in ComputationLayout()67 const ShapeLayout& result_layout() const { return result_layout_; } in result_layout() function
539 called_computation_layout.result_layout().shape(), instruction)); in AddMandatoryConstraints()568 if (body_layout.result_layout() != body_layout.parameter_layout(0)) { in AddMandatoryConstraints()571 << " shape=" << body_layout.result_layout().ToString(); in AddMandatoryConstraints()572 *body_layout.mutable_parameter_layout(0) = body_layout.result_layout(); in AddMandatoryConstraints()617 if (best_branch_computation_layout.result_layout() != in AddMandatoryConstraints()618 branch_computation_layout.result_layout()) { in AddMandatoryConstraints()628 << best_branch_computation_layout.result_layout().ToString(); in AddMandatoryConstraints()630 best_branch_computation_layout.result_layout(); in AddMandatoryConstraints()644 const ShapeLayout& result_layout = computation_layout->result_layout(); in AddMandatoryConstraints() local645 if (result_layout.LayoutIsSet()) { in AddMandatoryConstraints()[all …]
104 if (build_options.result_layout() != nullptr) { in CreateExecutionOptions()106 build_options.result_layout()->ToProto(); in CreateExecutionOptions()159 if (build_options.result_layout() != nullptr) { in CompileExecutable()160 TF_RETURN_IF_ERROR(ValidateResultShape(*build_options.result_layout(), in CompileExecutable()
945 const ShapeLayout& result_layout = layout.result_layout(); in VerifyEntryComputationLayout() local948 ShapeUtil::ValidateShapeWithOptionalLayout(result_layout.shape())); in VerifyEntryComputationLayout()950 TF_RETURN_IF_ERROR(VerifyNotSparse(result_layout.shape())); in VerifyEntryComputationLayout()953 result_layout.shape())) { in VerifyEntryComputationLayout()958 ShapeUtil::HumanString(result_layout.shape())); in VerifyEntryComputationLayout()1442 const Layout& result_layout = result_shape.layout(); in Postprocess() local1448 TF_RET_CHECK(LayoutUtil::Equal(result_layout, operand_layout)) in Postprocess()
45 const Shape* result_layout = nullptr; member
77 instance.result_layout->ToProto(); in CompileAheadOfTime()
297 module_config.entry_computation_layout().result_layout().shape(); in CreateFromProto()
244 negate->shape(), computation_layout.result_layout().shape())); in TEST_F()1305 if (!m->entry_computation_layout().result_layout().LayoutIsSet()) { in AssignLayoutsToComputation()
75 instance.result_layout = &result_shape; in CompileXla()