Home
last modified time | relevance | path

Searched refs:func_inputs (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dhash_utils_test.cc506 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/
Dunified_api_testutil.cc65 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()
Dgradients_util.cc251 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/
Dflatbuffer_import.cc937 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()