Searched refs:func_inputs (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/data/ |
D | hash_utils_test.cc | 506 std::vector<NodeDefBuilder::NodeOut> func_inputs; in TEST_F() local 507 func_inputs.emplace_back(n1->name(), 0, DT_FLOAT); in TEST_F() 508 func_inputs.emplace_back(n1->name(), 0, DT_FLOAT); in TEST_F() 515 .Input(func_inputs) in TEST_F() 529 .Input(func_inputs) in TEST_F() 571 std::vector<NodeDefBuilder::NodeOut> func_inputs; in TEST_F() local 572 func_inputs.emplace_back(n1->name(), 0, DT_FLOAT); in TEST_F() 573 func_inputs.emplace_back(n1->name(), 0, DT_FLOAT); in TEST_F() 580 .Input(func_inputs) in TEST_F() 595 .Input(func_inputs) in TEST_F() [all …]
|
/external/tensorflow/tensorflow/c/eager/ |
D | unified_api_testutil.cc | 65 std::vector<AbstractTensorHandle*> func_inputs; in RunModel() local 66 func_inputs.reserve(inputs.size()); in RunModel() 68 CreateParamsForInputs(func_ctx.get(), inputs, &func_inputs)); in RunModel() 71 TF_RETURN_IF_ERROR(model(func_ctx.get(), absl::MakeSpan(func_inputs), in RunModel() 73 for (auto func_input : func_inputs) { in RunModel()
|
D | gradients_util.cc | 251 vector<AbstractTensorHandle*> func_inputs; in RunModel() local 252 func_inputs.reserve(inputs.size()); in RunModel() 254 CreateParamsForInputs(func_ctx.get(), inputs, &func_inputs)); in RunModel() 257 TF_RETURN_IF_ERROR(model(func_ctx.get(), absl::MakeSpan(func_inputs), in RunModel() 259 for (auto func_input : func_inputs) { in RunModel()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/ |
D | flatbuffer_import.cc | 937 std::vector<int> func_inputs = subgraph.inputs; in ConvertSubgraph() local 944 TF_ASSIGN_OR_RETURN(func_inputs, in ConvertSubgraph() 948 for (int input : func_inputs) { in ConvertSubgraph() 980 const bool is_func_input = std::find(func_inputs.begin(), func_inputs.end(), in ConvertSubgraph() 981 output) != func_inputs.end(); in ConvertSubgraph() 1013 for (int i = 0, e = func_inputs.size(); i < e; i++) { in ConvertSubgraph() 1014 auto input_tensor = func_inputs[i]; in ConvertSubgraph() 1037 if (!func_inputs.empty()) { in ConvertSubgraph() 1039 subgraph, &builder, "inputs", func_inputs)); in ConvertSubgraph() 1053 PruneSubgraph(subgraph, func_inputs, func_outputs)); in ConvertSubgraph()
|