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

  /external/tensorflow/tensorflow/compiler/xla/service/
layout_assignment_test.cc 596 EXPECT_TRUE(ShapeUtil::TransposeIsBitcast(transpose->operand(0)->shape(),
614 EXPECT_TRUE(ShapeUtil::TransposeIsBitcast(transpose->operand(0)->shape(),
618 // TransposeIsBitcast shouldn't be called without layout information.
630 EXPECT_DEATH(ShapeUtil::TransposeIsBitcast(hlo->operand(0)->shape(),
656 // assignment, it should not call TransposeIsBitcast on the `transpose` node
657 // inside the fusion computation as TransposeIsBitcast checks both input_shape
    [all...]
algebraic_simplifier.cc 120 bool TransposeIsBitcast(const HloInstruction* transpose) {
123 return ShapeUtil::TransposeIsBitcast(operand->shape(), transpose->shape(),
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.cc 210 ShapeUtil::TransposeIsBitcast(operand_shape, shape, dimension_mapping)) {
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 543 // TransposeIsBitcast(shape, PermuteDimensions(permutation, shape),
586 static bool TransposeIsBitcast(const Shape& input_shape,
    [all...]
shape_util.cc     [all...]
shape_util_test.cc 704 // TransposeIsBitcast takes the inverse of the permutation that
706 EXPECT_TRUE(ShapeUtil::TransposeIsBitcast(
    [all...]

Completed in 184 milliseconds