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

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_instruction_fusion.cc 48 return !ShapeUtil::ElementIsComplex(hlo_shape) &&
dot_op_emitter.cc 495 if (ShapeUtil::ElementIsComplex(lhs_shape)) {
556 if (ShapeUtil::ElementIsComplex(lhs_array_.GetShape())) {
    [all...]
ir_emitter.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
instruction_fusion.cc 113 return ShapeUtil::ElementIsComplex(instruction.shape());
shape_inference.cc 262 !ShapeUtil::ElementIsComplex(shape)) {
271 if (ShapeUtil::ElementIsComplex(shape)) {
282 if (ShapeUtil::ElementIsComplex(shape)) {
304 !ShapeUtil::ElementIsComplex(shape)) {
313 !ShapeUtil::ElementIsComplex(shape)) {
    [all...]
elemental_ir_emitter.cc 213 } else if (ShapeUtil::ElementIsComplex(op->operand(0)->shape())) {
    [all...]
algebraic_simplifier.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.cc 490 if (ShapeUtil::ElementIsComplex(lhs_shape)) {
587 if (ShapeUtil::ElementIsComplex(lhs_shape)) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 444 static bool ElementIsComplex(const Shape& shape);
    [all...]
shape_util.cc 320 /* static */ bool ShapeUtil::ElementIsComplex(const Shape& shape) {
373 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape);
    [all...]
literal_comparison.cc 779 ShapeUtil::ElementIsComplex(expected.shape())) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
client_library_test_base.cc 318 ShapeUtil::ElementIsComplex(expected.shape())) {

Completed in 968 milliseconds