OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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