Searched refs:custom_call_has_side_effect (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | hlo_instructions.h | 1454 void set_custom_call_has_side_effect(bool custom_call_has_side_effect) { in set_custom_call_has_side_effect() argument 1455 custom_call_has_side_effect_ = custom_call_has_side_effect; in set_custom_call_has_side_effect() 1459 bool custom_call_has_side_effect() const { in custom_call_has_side_effect() function
|
D | hlo_instruction_test.cc | 1894 EXPECT_FALSE(custom_call_instr->custom_call_has_side_effect()); in TEST_F() 1896 EXPECT_TRUE(custom_call_instr->custom_call_has_side_effect()); in TEST_F() 1899 EXPECT_TRUE(custom_call_clone->custom_call_has_side_effect()); in TEST_F()
|
D | hlo_parser.cc | 2243 optional<bool> custom_call_has_side_effect; in ParseInstructionRhs() local 2261 &custom_call_has_side_effect}; in ParseInstructionRhs() 2357 if (custom_call_has_side_effect.has_value()) { in ParseInstructionRhs() 2359 *custom_call_has_side_effect); in ParseInstructionRhs()
|
D | hlo.proto | 233 bool custom_call_has_side_effect = 65; field
|
D | hlo_instruction.cc | 598 proto.custom_call_has_side_effect()); in CreateFromProto() 1541 ->custom_call_has_side_effect(); in HasSideEffectNoRecurse()
|
D | hlo_parser_test.cc | 1054 …,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 | 2547 casted_other.custom_call_has_side_effect()) { in IdenticalSlowPath()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | hlo_function_importer.cc | 325 builder_->getBoolAttr(custom_call->custom_call_has_side_effect()))); in ImportInstructionImpl()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/translate/ |
D | import.hlotxt | 293 …, f32[5,5] %arg2), custom_call_target="foo", backend_config="bar", custom_call_has_side_effect=true
|