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

  /external/tensorflow/tensorflow/compiler/xla/service/
algebraic_simplifier.h 39 bool ReshapeIsBitcast(const Shape& from_shape, const Shape& to_shape) const {
44 return ShapeUtil::ReshapeIsBitcast(from_shape, to_shape);
layout_assignment_test.cc 635 // ReshapeIsBitcast shouldn't be called without layout information.
648 ShapeUtil::ReshapeIsBitcast(hlo->operand(0)->shape(), hlo->shape()),
    [all...]
algebraic_simplifier.cc 133 if (options.ReshapeIsBitcast(operand->shape(), instr->shape())) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util_test.cc 664 ShapeUtil::ReshapeIsBitcast(
675 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(
734 EXPECT_FALSE(ShapeUtil::ReshapeIsBitcast(
747 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value()));
754 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value()));
764 EXPECT_TRUE(ShapeUtil::ReshapeIsBitcast(input, aligned_shape.value()));
    [all...]
shape_util.h 594 static bool ReshapeIsBitcast(const Shape& input_shape,
598 // ShapeUtil::ReshapeIsBitcast(input_shape, output_shape_with_layout) returns
    [all...]
shape_util.cc     [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
ir_array.cc 127 ShapeUtil::ReshapeIsBitcast(a, b);
174 ShapeUtil::ReshapeIsBitcast(input_shape, output_shape)) {
224 if (ShapeUtil::ReshapeIsBitcast(operand_shape, shape)) {

Completed in 481 milliseconds