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

Function Parameter

tensorflow/compiler/xla/service/pattern_matcher.h:2403–2406  ·  view source on GitHub ↗

Add overloads for Parameter which take an int64 specifying the parameter number.

Source from the content-addressed store, hash-verified

2401// Add overloads for Parameter which take an int64 specifying the parameter
2402// number.
2403inline auto Parameter(int64 parameter_num) -> decltype(
2404 Op().WithOpcode(HloOpcode::kParameter).WithParameterNum(parameter_num)) {
2405 return Op().WithOpcode(HloOpcode::kParameter).WithParameterNum(parameter_num);
2406}
2407template <typename HloInstructionType>
2408inline auto Parameter(HloInstructionType** matched_inst, int64 parameter_num)
2409 -> decltype(Op(matched_inst)

Callers

nothing calls this directly

Calls 3

WithParameterNumMethod · 0.80
WithOpcodeMethod · 0.80
OpFunction · 0.70

Tested by

no test coverage detected