Searched refs:variable_inputs (Results 1 – 4 of 4) sorted by relevance
94 std::vector<Value*> variable_inputs() const;
49 std::vector<Value*> GraphFloat32::variable_inputs() const { in variable_inputs() function in tflite::gpu::GraphFloat32
243 const auto variable_inputs = graph.variable_inputs(); in CopyInAndOutIds() local244 for (const auto& variable_input : variable_inputs) { in CopyInAndOutIds()
4301 std::vector<VariableInput> variable_inputs; in RewriteTPUReplicateNode() local4303 FindVariableInputs(*replicate_node, input_name_map, &variable_inputs)); in RewriteTPUReplicateNode()4309 control_before, control_after, variable_inputs, in RewriteTPUReplicateNode()4349 TF_RETURN_IF_ERROR(BuildVariableWrites(variable_inputs, control_after, in RewriteTPUReplicateNode()