Searched refs:AddDependency (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | value_inference.cc | 297 PostorderDFSNode& AddDependency(int64_t handle, PostorderDFSNodeType type, in AddDependency() function
484 result.AddDependency(operand_id, type, dep_context); in AnalyzeConstantValueFallback()
505 return result.AddDependency(caller_operand, type, context) in AnalyzeConstantValueFallback()
534 node.AddDependency(called_root, PostorderDFSNodeType::kConstantValue, in AnalyzeConstantValueFallback()
548 node.AddDependency(conditional_proto->operand_ids(0), in AnalyzeConstantValueFallback()
551 .AddDependency(conditional_proto->operand_ids(0), in AnalyzeConstantValueFallback()
563 node.AddDependency(branch_root, PostorderDFSNodeType::kConstantValue, in AnalyzeConstantValueFallback()
597 .AddDependency(operand_handle, type, context) in AnalyzeConstantValueFallback()
627 .AddDependency(root->operand_ids(tuple_operand_index), type, in AnalyzeConstantValueFallback()
668 .AddDependency(root->operand_ids(0), in AnalyzeUpperBound()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_future.h | 61 virtual void AddDependency(
245 event->AddDependency(promise_ref_.CopyRCRef()); in AssertHappensBefore()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | token_hlo_test.cc | 202 XLA_TEST_F(TokenHloTest, AddDependency) { in XLA_TEST_F() argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_matchers.h | 238 HLO_MATCHER(AddDependency);
|
| D | hlo_parser_test.cc | 2099 "AddDependency", in CreateTestCases()
2100 R"(HloModule AddDependency, entry_computation_layout={(f32[])->f32[]} in CreateTestCases()
2102 ENTRY AddDependency { in CreateTestCases()
|
| D | memory_space_assignment_test.cc | 1336 TEST_P(MemorySpaceAssignmentTest, AddDependency) { in TEST_P() argument
1361 op::Add(op::AddDependency(), op::Negate())); in TEST_P()
|
| D | hlo_dataflow_analysis_test.cc | 1964 TEST_P(HloDataflowAnalysisTest, AddDependency) { in TEST_P() argument
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/include/mlir-hlo/Dialect/mhlo/IR/ |
| D | hlo_ops.td | 859 let summary = "AddDependency operator";
861 AddDependency takes two operands: a data operand and a token. The output of
|