Searched refs:custom_call_has_side_effect (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instructions.h | 1857 void set_custom_call_has_side_effect(bool custom_call_has_side_effect) { in set_custom_call_has_side_effect() argument
1858 custom_call_has_side_effect_ = custom_call_has_side_effect; in set_custom_call_has_side_effect()
1862 bool custom_call_has_side_effect() const { in custom_call_has_side_effect() function
|
| D | hlo_instruction_test.cc | 2066 EXPECT_FALSE(custom_call_instr->custom_call_has_side_effect()); in TEST_F()
2068 EXPECT_TRUE(custom_call_instr->custom_call_has_side_effect()); in TEST_F()
2071 EXPECT_TRUE(custom_call_clone->custom_call_has_side_effect()); in TEST_F()
|
| D | hlo_parser.cc | 2589 optional<bool> custom_call_has_side_effect; in CreateInstruction() local
2610 &custom_call_has_side_effect}; in CreateInstruction()
2716 if (custom_call_has_side_effect.has_value()) { in CreateInstruction()
2718 *custom_call_has_side_effect); in CreateInstruction()
|
| D | hlo.proto | 288 bool custom_call_has_side_effect = 65; field
|
| D | hlo_instruction.cc | 769 proto.custom_call_has_side_effect()); in CreateFromProto()
1848 ->custom_call_has_side_effect(); in HasSideEffectNoRecurse()
|
| D | hlo_parser_test.cc | 1139 …,2,3]{0,1,2}) %p0, f32[123,4]{0,1} %p1), custom_call_target="baz", custom_call_has_side_effect=true in CreateTestCases()
|
| D | hlo_instructions.cc | 2861 casted_other.custom_call_has_side_effect()) { in IdenticalSlowPath()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
| D | import.hlotxt | 362 …, f32[5,5] %arg2), custom_call_target="foo", backend_config="bar", custom_call_has_side_effect=true
380 … s32[] %arg4), custom_call_target="foo", backend_config="bar", custom_call_has_side_effect=true, o…
396 …2[5,5] %arg2), custom_call_target="foo", backend_config="bar", custom_call_has_side_effect=true, o…
|
| D | export.mlir | 620 // CHECK-SAME: custom_call_has_side_effect=true
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 658 builder_->getBoolAttr(custom_call->custom_call_has_side_effect()))); in ImportInstructionImpl()
|