/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
D | split_test.cc | 28 outputs_.push_back(AddOutput(output)); in SplitOpModel() 42 return Dequantize<T>(ExtractVector<T>(outputs_[idx]), in GetDequantizedOutput() 43 GetScale(outputs_[idx]), GetZeroPoint(outputs_[idx])); in GetDequantizedOutput() 46 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape() 51 std::vector<int> outputs_; member in tflite::SplitOpModel
|
/external/tensorflow/tensorflow/core/grappler/ |
D | utils.h | 114 outputs_.reserve(graph->node_size()); in NodeMapInternal() 126 outputs_[NodeName(input)].insert(canonical); in NodeMapInternal() 135 auto it = outputs_.find(node_name); in GetOutputs() 136 if (it == outputs_.end()) { in GetOutputs() 146 auto it = outputs_.find(node_name); in GetOutputsOrderedByNodeName() 147 if (it != outputs_.end()) { in GetOutputsOrderedByNodeName() 171 outputs_.erase(NodeName(name)); in RemoveNode() 193 outputs_[node_name].insert(output_node); in AddOutput() 197 outputs_[node_name].erase(nodes_[NodeName(output_name)]); in RemoveOutput() 213 void RemoveOutputs(const string& node_name) { outputs_.erase(node_name); } in RemoveOutputs() [all …]
|
/external/tensorflow/tensorflow/lite/experimental/kernels/ |
D | ctc_beam_search_decoder_test.cc | 51 outputs_.push_back(AddOutput(TensorType_INT32)); in CTCBeamSearchDecoderOpModel() 53 outputs_.push_back(AddOutput(TensorType_FLOAT32)); in CTCBeamSearchDecoderOpModel() 73 for (int i = 0; i < outputs_.size() - 1; ++i) { in GetDecodedOutpus() 74 outputs.push_back(ExtractVector<int>(outputs_[i])); in GetDecodedOutpus() 80 return ExtractVector<float>(outputs_[outputs_.size() - 1]); in GetLogProbabilitiesOutput() 85 for (const int output : outputs_) { in GetOutputShapes() 94 std::vector<int> outputs_; member in tflite::ops::experimental::__anonc74d09040111::CTCBeamSearchDecoderOpModel
|
/external/tensorflow/tensorflow/lite/delegates/flex/ |
D | kernel.cc | 127 outputs_.push_back(index); in OpOutputs() 129 vector_.resize(outputs_.size()); in OpOutputs() 137 for (int i : outputs_) { in InitializeGraphOutputs() 160 int Size() const { return outputs_.size(); } in Size() 162 int TfLiteIndex(int i) const { return outputs_[i]; } in TfLiteIndex() 180 std::vector<int> outputs_; member in tflite::flex::OpOutputs 190 : inputs_(inputs), outputs_(outputs) {} in OpNode() 209 const OpOutputs& outputs() const { return outputs_; } in outputs() 210 OpOutputs* mutable_outputs() { return &outputs_; } in mutable_outputs() 213 int NumOutputs() const { return outputs_.Size(); } in NumOutputs() [all …]
|
/external/tensorflow/tensorflow/c/ |
D | while_loop_test.cc | 57 outputs_.resize(ninputs, {nullptr, -1}); in Init() 61 TF_FinishWhile(params_.get(), s_, &outputs_[0]); in ExpectOK() 66 TF_FinishWhile(params_.get(), s_, &outputs_[0]); in ExpectError() 75 Run(outputs_, input_values); in Run() 128 std::vector<TF_Output> outputs_; // The final outputs of the while loop member in __anonbc36f9730111::CApiWhileLoopTest 177 EXPECT_TRUE(outputs_[0].oper != nullptr); in TEST_F() 178 EXPECT_GE(outputs_[0].index, 0); in TEST_F() 179 EXPECT_TRUE(outputs_[1].oper != nullptr); in TEST_F() 180 EXPECT_GE(outputs_[1].index, 0); in TEST_F() 433 TF_AddGradients(graph_, outputs_.data(), outputs_.size(), inputs_.data(), 1, in TEST_F()
|
D | c_test_util.cc | 481 outputs_.clear(); in SetOutputs() 483 outputs_.emplace_back(TF_Output{o, 0}); in SetOutputs() 485 output_values_.resize(outputs_.size()); in SetOutputs() 490 outputs_ = outputs; in SetOutputs() 491 output_values_.resize(outputs_.size()); in SetOutputs() 507 output_values_.resize(outputs_.size(), nullptr); in Run() 513 const TF_Output* outputs_ptr = outputs_.empty() ? nullptr : &outputs_[0]; in Run() 520 outputs_ptr, output_values_ptr, outputs_.size(), targets_ptr, in Run()
|
/external/tensorflow/tensorflow/cc/ops/ |
D | while_loop_test.cc | 43 ops::BuildWhileLoop(scope_, inputs_, cond, body, kFrameName, &outputs_); in CreateLoop() 60 TF_ASSERT_OK(session.Run(feeds, outputs_, &out_tensors)); in Run() 61 ASSERT_EQ(out_tensors.size(), outputs_.size()); in Run() 72 std::vector<Output> outputs_; member in tensorflow::__anon7e249d170111::WhileLoopTest 98 for (int i = 0; i < outputs_.size(); ++i) { in TEST_F() 99 Node* node = outputs_[i].node(); in TEST_F()
|
/external/tensorflow/tensorflow/cc/framework/ |
D | while_gradients_test.cc | 45 &outputs_)); in CreateLoop() 50 AddSymbolicGradients(scope_, outputs_, inputs_, &grad_outputs_)); in CreateBackprop() 87 std::vector<Output> outputs_; member in tensorflow::__anon97a6b1da0111::WhileGradientsTest 182 outputs_[0] = ops::Neg(scope_, outputs_[0]); in TEST_F() 213 AddSymbolicGradients(cpu1_scope, outputs_, inputs_, &grad_outputs_)); in TEST_F()
|
D | gradients.cc | 104 const std::vector<Output>& outputs_; member in tensorflow::__anonc8091ec30111::SymbolicGradientBuilder 147 outputs_(outputs), in SymbolicGradientBuilder() 171 for (const Output& out : outputs_) { in GetReachableNodes() 251 if (outputs_.size() != grad_inputs_.size()) { in Initialize() 268 output_nodes.reserve(outputs_.size()); in Initialize() 269 for (size_t i = 0; i < outputs_.size(); ++i) { in Initialize() 270 output_nodes.insert(outputs_[i].node()->id()); in Initialize() 323 for (const Output& output : outputs_) { in Initialize() 337 TF_RETURN_IF_ERROR(BackpropAlongEdge(grad_inputs_[i], outputs_[i])); in Initialize()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | split_v_test.cc | 45 outputs_.push_back(AddOutput(input.type)); in SplitVOpModel() 68 return ExtractVector<T>(outputs_[i]); in GetOutput() 70 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape() 76 std::vector<int> outputs_; member in tflite::__anon3edbe8b00111::SplitVOpModel
|
D | unpack_test.cc | 42 outputs_.push_back(AddOutput(input.type)); in UnpackOpModel() 55 for (const int output : outputs_) { in GetOutputDatas() 64 for (const int output : outputs_) { in GetOutputShapes() 72 std::vector<int> outputs_; member in tflite::__anona011a1f10111::UnpackOpModel
|
D | split_test.cc | 50 outputs_.push_back(AddOutput(input.type)); in SplitOpModel() 69 return ExtractVector<T>(outputs_[i]); in GetOutput() 71 std::vector<int> GetOutputShape(int i) { return GetTensorShape(outputs_[i]); } in GetOutputShape() 76 std::vector<int> outputs_; member in tflite::__anon363ae4fb0111::SplitOpModel
|
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
D | api.cc | 465 return LinkTensors(outputs, factory, &outputs_); in Initialize() 473 return GetExternalDefinitions(outputs_); in outputs() 485 if (index < 0 || index >= outputs_.size()) { in GetOutputObject() 488 *object = outputs_[index]->GetExternalObject(); in GetOutputObject() 500 if (index < 0 || index >= outputs_.size()) { in SetOutputObject() 503 return outputs_[index]->SetExternalObject(object); in SetOutputObject() 515 if (index > outputs_.size()) { in CopyToExternalOutput() 519 return outputs_[index]->CopyToExternalObject(); in CopyToExternalOutput() 534 for (int i = 0; i < outputs_.size(); i++) { in Run() 581 std::vector<std::unique_ptr<TensorTie>> outputs_; member in tflite::gpu::cl::__anon117e98fd0111::InferenceRunnerImpl [all …]
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | lower_case_op.cc | 67 std::vector<NodeOut> outputs_; member in tensorflow::__anon74dd97450111::CaseBuilder 202 outputs_.resize(merges.size()); in AddOutputs() 214 outputs_[i] = NodeOut(merges[i], 0); in AddOutputs() 260 NodeBuilder builder = keep_node_fetchable_ && !outputs_.empty() in BuildLoweredCaseOutput() 261 ? NodeBuilder(name_, "IdentityN").Input(outputs_) in BuildLoweredCaseOutput()
|
D | lower_if_op.cc | 74 std::vector<NodeOut> outputs_; member in tensorflow::__anon5deaa8580111::CondBuilder 237 outputs_.resize(merges.size()); in AddOutputs() 245 outputs_[i] = NodeOut(merges[i], 0); in AddOutputs() 289 NodeBuilder builder = keep_node_fetchable_ && !outputs_.empty() in BuildLoweredIfOutput() 290 ? NodeBuilder(name_, "IdentityN").Input(outputs_) in BuildLoweredIfOutput()
|
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/ |
D | cluster.cc | 44 outputs_.push_back(node.input(i)); in FindClusterInputsAndOutputs() 49 return (!inputs_.empty()) && (!outputs_.empty()); in FindClusterInputsAndOutputs()
|
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
D | test_util.h | 57 return outputs_[index].data; in GetOutput() 63 std::vector<TensorFloat32> outputs_; variable
|
/external/tensorflow/tensorflow/lite/ |
D | arena_planner_test.cc | 45 : inputs_(inputs), outputs_(outputs), temporaries_(temporaries) {} in TestOp() 48 const std::vector<int>& outputs() const { return outputs_; } in outputs() 53 std::vector<int> outputs_; member in tflite::__anond3fd21620111::TestOp 64 : inputs_(inputs), outputs_(outputs) { in TestGraph() 115 const std::vector<int>& outputs() { return outputs_; } in outputs() 126 std::swap(outputs_, other->outputs_); in Swap() 134 std::vector<int> outputs_; member in tflite::__anond3fd21620111::TestGraph
|
/external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
D | tflite_inference_stage.h | 74 const std::vector<void*>* GetOutputs() const { return &outputs_; } in GetOutputs() 87 std::vector<void*> outputs_; variable
|
D | tflite_inference_stage.cc | 46 outputs_.clear(); in UpdateModelInfo() 47 outputs_.reserve(interpreter_->outputs().size()); in UpdateModelInfo() 50 outputs_.push_back(tensor->data.raw); in UpdateModelInfo()
|
/external/tensorflow/tensorflow/core/framework/ |
D | op_kernel.cc | 330 : params_(params), outputs_(num_outputs) { in OpKernelContext() 348 for (TensorValue& value : outputs_) { in ~OpKernelContext() 476 outputs_[output_index] = TensorValue(new_tensor.release()); in forward_input_to_output_with_shape() 477 *output = outputs_[output_index].tensor; in forward_input_to_output_with_shape() 739 " num_outputs=", outputs_.size(), in allocate_output() 769 outputs_[index] = TensorValue(output_tensor.release()); in allocate_output() 770 *output = outputs_[index].tensor; in allocate_output() 942 outputs_[index] = TensorValue(new_tensor.release()); in maybe_set_output_by_allocate_and_copy() 968 CHECK_LT(index, outputs_.size()); in set_output() 971 CHECK_EQ(outputs_[index].tensor, nullptr); in set_output() [all …]
|
/external/tensorflow/tensorflow/lite/tools/serialization/ |
D | writer_lib.h | 91 outputs_(subgraph->outputs()), in SubgraphWriter() 126 outputs_(subgraph->outputs()), in SubgraphWriter() 189 std::vector<int> outputs_; variable
|
/external/tensorflow/tensorflow/lite/delegates/gpu/ |
D | gl_delegate.cc | 212 outputs_.clear(); in Prepare() 213 outputs_.reserve(delegate_params->output_tensors->size); in Prepare() 224 outputs_.push_back(output->id); in Prepare() 312 for (ValueId id : outputs_) { in Invoke() 364 std::vector<ValueId> outputs_; member in tflite::gpu::gl::__anon640f09d20111::Delegate
|
/external/tensorflow/tensorflow/java/src/gen/cc/ |
D | op_specs.h | 148 const std::vector<ArgumentSpec>& outputs() const { return outputs_; } in outputs() 171 std::vector<ArgumentSpec> outputs_; variable
|
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
D | api2.cc | 507 outputs_ = LinkTensors(graph_.outputs()); in Initialize() 516 return GetExternalDefinitions(outputs_); in outputs() 541 if (index < 0 || index >= outputs_.size()) { in SetOutputObjectDef() 544 auto def = outputs_[index]; in SetOutputObjectDef() 550 outputs_[index] = def; in SetOutputObjectDef() 582 RETURN_IF_ERROR(runner_impl->Initialize(inputs_, outputs_, &tie_factory_)); in Build() 655 std::vector<TensorTieDef> outputs_; member in tflite::gpu::gl::__anon517d4d930111::InferenceBuilderImpl
|