Searched refs:layout_sensitive (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_verifier.h | 32 ShapeVerifier(bool layout_sensitive, bool allow_mixed_precision, in ShapeVerifier() argument
34 : layout_sensitive_(layout_sensitive), in ShapeVerifier()
239 bool layout_sensitive, bool allow_mixed_precision, in DefaultVerifierMetadata() argument
242 layout_sensitive_(layout_sensitive), in DefaultVerifierMetadata()
266 bool layout_sensitive, bool allow_mixed_precision,
272 layout_sensitive, allow_mixed_precision, shape_size_func)), in target_metadata_()
275 CHECK(instruction_can_change_layout_func_ == nullptr || layout_sensitive); in target_metadata_()
|
| D | hlo_instruction.h | 1143 bool layout_sensitive = true) const {
1145 layout_sensitive,
1157 bool layout_sensitive = true) const {
1159 layout_sensitive,
1998 bool layout_sensitive, bool ignore_channel_id_values) const;
|
| D | hlo_instruction.cc | 2017 bool layout_sensitive, bool ignore_channel_id_values) const { in IdenticalInternal() argument
2028 if (!(layout_sensitive ? ShapeUtil::Equal(shape(), other.shape()) in IdenticalInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | test_utils.h | 106 Status VerifyHloModule(HloModule* const module, bool layout_sensitive,
|
| D | test_utils.cc | 706 Status VerifyHloModule(HloModule* const module, bool layout_sensitive, in VerifyHloModule() argument
708 return HloVerifier(/*layout_sensitive=*/layout_sensitive, in VerifyHloModule()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_helpers.inc | 376 auto layout_sensitive = cast<LayoutSensitiveInterface>(op->getOperation());
377 for (unsigned idx : layout_sensitive.GetLayoutDependentResults()) {
|