Home
last modified time | relevance | path

Searched refs:expected_program_shape (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module.cc282 ProgramShape expected_program_shape(proto.host_program_shape()); in CreateFromProto() local
283 TF_RET_CHECK(expected_program_shape.parameters_size() == in CreateFromProto()
285 for (int i = 0; i < expected_program_shape.parameters_size(); ++i) { in CreateFromProto()
288 TF_RET_CHECK(ShapeUtil::Compatible(expected_program_shape.parameters(i), in CreateFromProto()
293 expected_program_shape.parameters(i)) in CreateFromProto()
299 ShapeUtil::Compatible(expected_program_shape.result(), result_shape)) in CreateFromProto()
302 << ShapeUtil::HumanStringWithLayout(expected_program_shape.result()) in CreateFromProto()