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

  /external/tensorflow/tensorflow/compiler/xla/service/
shape_inference.h 65 static StatusOr<Shape> InferTernaryOpShape(HloOpcode opcode, const Shape& lhs,
68 static StatusOr<Shape> InferTernaryOpShape(HloOpcode opcode,
shape_inference_test.cc 113 auto inferred_status = ShapeInference::InferTernaryOpShape(
120 auto inferred_status = ShapeInference::InferTernaryOpShape(
128 auto inferred_status = ShapeInference::InferTernaryOpShape(
135 auto inferred_status_error1 = ShapeInference::InferTernaryOpShape(
141 auto inferred_status_error2 = ShapeInference::InferTernaryOpShape(
147 auto inferred_status_error3 = ShapeInference::InferTernaryOpShape(
155 auto inferred_status_error4 = ShapeInference::InferTernaryOpShape(
165 auto inferred_status = ShapeInference::InferTernaryOpShape(
173 ShapeInference::InferTernaryOpShape(HloOpcode::kClamp, f32_, f32_, f32_);
179 auto inferred_status = ShapeInference::InferTernaryOpShape(
    [all...]
hlo_creation_utils.cc 282 ShapeInference::InferTernaryOpShape(
shape_inference.cc     [all...]
hlo_verifier.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/
xla_builder.cc 561 Shape shape, ShapeInference::InferTernaryOpShape(triop, lhs_shape,
    [all...]

Completed in 180 milliseconds