MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / HandleSetDimensionSize

Method HandleSetDimensionSize

tensorflow/compiler/xla/service/hlo_evaluator.cc:429–438  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

427}
428
429Status HloEvaluator::HandleSetDimensionSize(
430 HloInstruction* set_dimension_size) {
431 const Literal& operand_literal =
432 GetEvaluatedLiteralFor(set_dimension_size->operand(0));
433 Literal result(set_dimension_size->shape());
434 memcpy(result.untyped_data(), operand_literal.untyped_data(),
435 operand_literal.size_bytes());
436 evaluated_[set_dimension_size] = std::move(result);
437 return Status::OK();
438}
439
440Status HloEvaluator::HandleParameter(HloInstruction* parameter) {
441 // Nothing to do other than sanity checks. Parameters' values are stored in

Callers

nothing calls this directly

Calls 4

untyped_dataMethod · 0.80
operandMethod · 0.45
shapeMethod · 0.45
size_bytesMethod · 0.45

Tested by

no test coverage detected