Searched refs:const_inputs (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/ |
D | eval_const_tensor.cc | 130 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/ |
D | function_optimizer.cc | 185 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 …]
|
D | constant_folding.cc | 3083 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 …]
|
D | remapper.cc | 447 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/ |
D | vbo_exec_array.c | 302 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()
|