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

  /external/tensorflow/tensorflow/compiler/xla/service/
ar_crs_combiner.cc 55 return ShapeUtil::ElementIsFloating(instruction->shape()) ==
56 ShapeUtil::ElementIsFloating(instruction->operand(0)->shape());
61 return ShapeUtil::ElementIsFloating(instruction->shape());
shape_inference.cc 245 if (!ShapeUtil::ElementIsFloating(shape)) {
261 if (!ShapeUtil::ElementIsFloating(shape) &&
273 } else if (ShapeUtil::ElementIsFloating(shape)) {
303 !ShapeUtil::ElementIsFloating(shape) &&
332 if (!ShapeUtil::ElementIsFloating(shape)) {
463 if (!ShapeUtil::ElementIsFloating(operand_shape)) {
    [all...]
hlo_verifier.cc 789 if (!ShapeUtil::ElementIsFloating(subshape)) {
    [all...]
hlo_evaluator_typed_visitor.h     [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 258 if (ElementIsFloating(a) && ElementIsFloating(b)) {
441 static bool ElementIsFloating(const Shape& shape);
    [all...]
literal_comparison.cc 778 if (ShapeUtil::ElementIsFloating(expected.shape()) ||
    [all...]
shape_util.cc 324 /* static */ bool ShapeUtil::ElementIsFloating(const Shape& shape) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
instruction_fusion.cc 48 return !ShapeUtil::ElementIsFloating(instruction.shape()) &&
ir_emitter.cc 495 } else if (ShapeUtil::ElementIsFloating(lhs_shape)) {
595 } else if (ShapeUtil::ElementIsFloating(lhs_shape)) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
client_library_test_base.cc 317 if (ShapeUtil::ElementIsFloating(expected.shape()) ||
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.cc     [all...]

Completed in 329 milliseconds