Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel.h94 std::vector<Value*> variable_inputs() const;
Dgpu_model.cc683 const auto variable_inputs = graph.variable_inputs(); in CopyExternals() local
684 for (const auto& value : variable_inputs) { in CopyExternals()
Dmodel.cc49 std::vector<Value*> GraphFloat32::variable_inputs() const { in variable_inputs() function in tflite::gpu::GraphFloat32
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Ddistributed_tpu_rewrite_pass.cc4803 std::vector<VariableInput> variable_inputs; in RewriteTPUReplicateNode() local
4805 FindVariableInputs(*replicate_node, input_name_map, &variable_inputs)); in RewriteTPUReplicateNode()
4811 control_before, control_after, variable_inputs, in RewriteTPUReplicateNode()
4854 TF_RETURN_IF_ERROR(BuildVariableWrites(variable_inputs, control_after, in RewriteTPUReplicateNode()