HomeSort by relevance Sort by last modified time
    Searched refs:InferBinaryOpShape (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
shape_inference_test.cc 259 return ShapeInference::InferBinaryOpShape(BinaryOperation::BINOP_COMPLEX,
    [all...]
shape_inference.h 56 static StatusOr<Shape> InferBinaryOpShape(
59 static StatusOr<Shape> InferBinaryOpShape(HloOpcode opcode,
algebraic_simplifier.cc 386 ShapeInference::InferBinaryOpShape(HloOpcode::kAdd, c1, c2));
624 ShapeInference::InferBinaryOpShape(HloOpcode::kMultiply,
631 ShapeInference::InferBinaryOpShape(HloOpcode::kMultiply,
644 ShapeInference::InferBinaryOpShape(
657 ShapeInference::InferBinaryOpShape(
    [all...]
shape_inference.cc     [all...]
hlo_verifier.cc 521 instruction, ShapeInference::InferBinaryOpShape(instruction->opcode(),
    [all...]
user_computation.cc     [all...]

Completed in 552 milliseconds