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/
ir_emission_utils.cc 43 if (ShapeUtil::ElementIsComplex(input_shape) ||
44 ShapeUtil::ElementIsComplex(kernel_shape)) {
dot_op_emitter.cc     [all...]
ir_emitter.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
client_library_test_base.cc 267 ShapeUtil::ElementIsComplex(expected.shape())) {
324 ShapeUtil::ElementIsComplex(expected.shape()));
literal_test_util.cc 713 ShapeUtil::ElementIsComplex(expected.shape())) {
  /external/tensorflow/tensorflow/compiler/xla/service/
instruction_fusion.cc 88 return ShapeUtil::ElementIsComplex(instruction.shape());
shape_inference.cc 334 !ShapeUtil::ElementIsComplex(arg)) {
343 if (!ShapeUtil::ElementIsComplex(arg)) {
351 if (ShapeUtil::ElementIsComplex(arg)) {
    [all...]
algebraic_simplifier.cc     [all...]
elemental_ir_emitter.cc 209 } else if (ShapeUtil::ElementIsComplex(op->operand(0)->shape())) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 370 static bool ElementIsComplex(const Shape& shape);
shape_util.cc 312 /* static */ bool ShapeUtil::ElementIsComplex(const Shape& shape) {
361 CHECK(ElementIsComplex(complex_shape)) << HumanString(complex_shape);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter.cc 458 if (ShapeUtil::ElementIsComplex(lhs_shape)) {
550 if (ShapeUtil::ElementIsComplex(lhs_shape)) {
    [all...]

Completed in 146 milliseconds