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

Method InferUnaryOpShape

tensorflow/compiler/xla/service/shape_inference.cc:225–228  ·  view source on GitHub ↗

static */

Source from the content-addressed store, hash-verified

223} // namespace
224
225/* static */ StatusOr<Shape> ShapeInference::InferUnaryOpShape(
226 HloOpcode opcode, const HloInstruction* operand) {
227 return InferUnaryOpShape(opcode, operand->shape());
228}
229
230/* static */ StatusOr<Shape> ShapeInference::InferUnaryOpShape(
231 HloOpcode opcode, const Shape& shape) {

Callers

nothing calls this directly

Calls 7

ExpectArrayFunction · 0.85
InvalidArgumentFunction · 0.85
HloOpcodeStringFunction · 0.85
ComplexComponentTypeFunction · 0.85
IsIntegralTypeFunction · 0.85
shapeMethod · 0.45
element_typeMethod · 0.45

Tested by

no test coverage detected