| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_test.cc | 36 ASSERT_TRUE(graph.SetProducer(node->id, graph_output->id).ok()); in TEST()
58 ASSERT_TRUE(graph.SetProducer(node->id, graph_output1->id).ok()); in TEST()
59 ASSERT_TRUE(graph.SetProducer(node->id, graph_output2->id).ok()); in TEST()
104 ASSERT_TRUE(graph.SetProducer(node->id, graph_output->id).ok()); in TEST()
105 EXPECT_FALSE(graph.SetProducer(node->id, graph_output->id).ok()); in TEST()
128 ASSERT_TRUE(graph.SetProducer(node->id, graph_output->id).ok()); in TEST()
147 ASSERT_TRUE(graph_.SetProducer(node_->id, graph_output->id).ok()); in SetUp()
180 ASSERT_TRUE(graph_.SetProducer(first_node_->id, value_->id).ok()); in SetUp()
182 ASSERT_TRUE(graph_.SetProducer(second_node_->id, graph_output_->id).ok()); in SetUp()
235 ASSERT_TRUE(graph_.SetProducer(first_node_->id, value0_->id).ok()); in SetUp()
[all …]
|
| D | lstm_parser.cc | 138 RETURN_IF_ERROR(graph->SetProducer(node->id, input_times_weights->id)); in BuildLstmGate()
152 graph->SetProducer(node->id, output_state_times_weights->id)); in BuildLstmGate()
167 RETURN_IF_ERROR(graph->SetProducer(node->id, cell_state_times_weights->id)); in BuildLstmGate()
183 graph->SetProducer(add_node->id, gate_before_normalization->id)); in BuildLstmGate()
202 RETURN_IF_ERROR(graph->SetProducer(node->id, normalized_gate->id)); in BuildLstmGate()
217 graph->SetProducer(node->id, reweighted_normalized_gate->id)); in BuildLstmGate()
231 RETURN_IF_ERROR(graph->SetProducer(node->id, gate->id)); in BuildLstmGate()
264 RETURN_IF_ERROR(graph->SetProducer(node->id, cell_state_contrib->id)); in BuildCellStateUpdate()
274 RETURN_IF_ERROR(graph->SetProducer(node->id, cell_gate_contrib->id)); in BuildCellStateUpdate()
283 RETURN_IF_ERROR(graph->SetProducer(node->id, new_cell_state->id)); in BuildCellStateUpdate()
[all …]
|
| D | model.cc | 155 absl::Status GraphFloat32::SetProducer(NodeId producer, ValueId value) { in SetProducer() function in tflite::gpu::GraphFloat32
331 RETURN_IF_ERROR(model->SetProducer(node_def.node->id, output->id)); in MakeExactCopy()
414 RETURN_IF_ERROR(graph->SetProducer(to_keep->id, output->id)); in RemoveFollowingNode()
462 RETURN_IF_ERROR(graph->SetProducer(producer->id, output_id)); in RemoveSimpleNodeKeepOutput()
477 RETURN_IF_ERROR(graph->SetProducer(from_node->id, link->id)); in AddOutput()
|
| D | object_reader.cc | 147 RETURN_IF_ERROR(graph_->SetProducer(node->id, value->id)); in AddOutput()
191 RETURN_IF_ERROR(graph_->SetProducer(node->id, updated_value->id)); in AddUpdate()
|
| D | model_builder_helper.cc | 57 RETURN_IF_ERROR(graph->SetProducer((*passthru_node)->id, output->id)); in NewPassthroughNode()
59 RETURN_IF_ERROR(graph->SetProducer(node->id, copy_output->id)); in NewPassthroughNode()
|
| D | model.h | 138 absl::Status SetProducer(NodeId producer, ValueId value);
|
| D | model_builder.cc | 109 RETURN_IF_ERROR(graph->SetProducer(node->id, (*value)->id)); in NewConstNode()
834 RETURN_IF_ERROR(graph->SetProducer(reshape->id, reshaped_value->id)); in Parse()
989 RETURN_IF_ERROR(graph->SetProducer(concat_node->id, concat_temp->id)); in ParseBasic()
992 RETURN_IF_ERROR(graph->SetProducer(fc_node->id, activ_temp->id)); in ParseBasic()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/ |
| D | fuse_auto_input_test.cc | 46 ASSERT_OK(graph.SetProducer(n1->id, v1->id)); in TEST()
52 ASSERT_OK(graph.SetProducer(n2->id, v2->id)); in TEST()
59 ASSERT_OK(graph.SetProducer(n3->id, v3->id)); in TEST()
83 ASSERT_OK(graph.SetProducer(n1->id, v1->id)); in TEST()
90 ASSERT_OK(graph.SetProducer(n2->id, v2->id)); in TEST()
|
| /external/dagger2/javatests/dagger/producers/internal/ |
| D | SetProducerTest.java | 41 SetProducer.<Integer>builder(1, 1) in success()
51 SetProducer.<Integer>builder(1, 0) in delegateNpe()
66 SetProducer.<Integer>builder(0, 1) in delegateSetNpe()
81 SetProducer.<Integer>builder(0, 2) in delegateElementNpe()
|
| /external/dagger2/java/dagger/producers/internal/ |
| D | SetProducer.java | 40 public final class SetProducer<T> extends AbstractProducer<Set<T>> { class
87 public SetProducer<T> build() { in build()
93 return new SetProducer<T>(individualProducers, collectionProducers); in build()
100 private SetProducer( in SetProducer() method in SetProducer
|
| D | SetOfProducedProducer.java | 43 return SetProducer.empty(); in empty()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | merge_padding_with_test.cc | 134 ASSERT_TRUE(graph.SetProducer(pad_node->id, padded->id).ok()); in TEST()
140 ASSERT_TRUE(graph.SetProducer(add_node->id, output->id).ok()); in TEST()
172 ASSERT_TRUE(graph.SetProducer(pad_node->id, padded->id).ok()); in TEST()
179 ASSERT_TRUE(graph.SetProducer(add_node->id, output->id).ok()); in TEST()
|
| D | remove_noop_test.cc | 226 ASSERT_TRUE(graph.SetProducer(producer_node->id, value0->id).ok()); in TEST()
228 ASSERT_TRUE(graph.SetProducer(simple_node->id, value1->id).ok()); in TEST()
230 ASSERT_TRUE(graph.SetProducer(consumer_node->id, graph_output->id).ok()); in TEST()
|
| D | add_quant_adjustments.cc | 74 graph->SetProducer(quant_and_dequant_node->id, adjusted_value->id); in ApplyToNode()
|
| /external/dagger2/java/dagger/internal/codegen/javapoet/ |
| D | TypeNames.java | 46 import dagger.producers.internal.SetProducer;
90 public static final ClassName SET_PRODUCER = ClassName.get(SetProducer.class);
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | test_util.cc | 58 graph_.SetProducer(node->id, output->id).IgnoreError(); in SingleOpModel()
|
| /external/dagger2/java/dagger/internal/codegen/binding/ |
| D | SourceFiles.java | 62 import dagger.producers.internal.SetProducer;
|