Home
last modified time | relevance | path

Searched refs:const_inputs (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.cc130 std::vector<std::pair<string, Tensor>>* const_inputs) { in ExtractConstantSubgraph() argument
276 const_inputs->emplace_back(output_tensor_name, tensor_inferred); in ExtractConstantSubgraph()
287 const_inputs->emplace_back(output_tensor_name, it->second); in ExtractConstantSubgraph()
336 std::vector<std::pair<string, Tensor>> const_inputs; in EvaluateConstantTensor() local
339 &const_inputs)); in EvaluateConstantTensor()
357 const_inputs, {output_tensor_name}, &outputs); in EvaluateConstantTensor()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc185 absl::flat_hash_map<InputPort, string> const_inputs; member
192 const_inputs == other.const_inputs; in operator ==()
218 + s.const_inputs.size() * 2); in AbslHashValue()
238 absl::c_for_each(s.const_inputs, [&hashes](const ConstInput& const_input) { in AbslHashValue()
254 absl::flat_hash_set<string> const_inputs; member
553 absl::flat_hash_set<string>* const_inputs, in PushDownConstInputs() argument
574 const_inputs->insert(input); in PushDownConstInputs()
588 if (specialization.const_inputs.empty()) return; in RemovePushedDownConstInputs()
595 return specialization.const_inputs.find(input) == in RemovePushedDownConstInputs()
596 specialization.const_inputs.end(); in RemovePushedDownConstInputs()
[all …]
Dconstant_folding.cc3083 std::vector<int> const_inputs; in PartialAssocOpConstFolding() local
3090 const_inputs.push_back(i); in PartialAssocOpConstFolding()
3098 if (const_inputs.size() == num_non_control_inputs && in PartialAssocOpConstFolding()
3105 *node, strings::StrCat("_partial_split_", const_inputs.size())); in PartialAssocOpConstFolding()
3106 if (1 < const_inputs.size() && in PartialAssocOpConstFolding()
3107 const_inputs.size() < num_non_control_inputs && in PartialAssocOpConstFolding()
3118 for (int i : const_inputs) { in PartialAssocOpConstFolding()
3125 node->set_input(const_inputs[0], added_node->name()); in PartialAssocOpConstFolding()
3127 nonconst_inputs.push_back(const_inputs[0]); in PartialAssocOpConstFolding()
3138 const_inputs.size() - 1); in PartialAssocOpConstFolding()
[all …]
Dremapper.cc447 auto const_inputs = std::count_if( in FindFusedBatchNorm() local
453 bool can_remap = const_scaling_factor || const_inputs >= 4; in FindFusedBatchNorm()
/external/mesa3d/src/mesa/vbo/
Dvbo_exec_array.c302 GLbitfield const_inputs = 0x0; in recalculate_input_bindings() local
317 const_inputs |= VERT_BIT(i); in recalculate_input_bindings()
324 const_inputs |= VERT_BIT_GENERIC(i); in recalculate_input_bindings()
333 const_inputs |= VERT_BIT_GENERIC(i); in recalculate_input_bindings()
360 const_inputs |= VERT_BIT_POS; in recalculate_input_bindings()
368 const_inputs |= VERT_BIT_FF(i); in recalculate_input_bindings()
379 const_inputs |= VERT_BIT_GENERIC(i); in recalculate_input_bindings()
394 const_inputs |= VERT_BIT_FF(i); in recalculate_input_bindings()
404 const_inputs |= VERT_BIT_GENERIC(i); in recalculate_input_bindings()
412 _mesa_set_varying_vp_inputs(ctx, VERT_BIT_ALL & (~const_inputs)); in recalculate_input_bindings()