Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Deval_const_tensor.cc145 std::vector<std::pair<string, Tensor>>* const_inputs, in ExtractConstantSubgraph() argument
254 const_inputs->emplace_back(tensor_name, *const_tensor); in ExtractConstantSubgraph()
316 const_inputs->emplace_back(output_tensor_name, tensor_inferred); in ExtractConstantSubgraph()
327 const_inputs->emplace_back(output_tensor_name, it->second); in ExtractConstantSubgraph()
393 std::vector<std::pair<string, Tensor>> const_inputs; in EvaluateConstantTensor() local
396 &const_inputs, outer_context)); in EvaluateConstantTensor()
414 const_inputs, {output_tensor_name}, &outputs); in EvaluateConstantTensor()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc168 absl::flat_hash_map<InputPort, string> const_inputs; member
175 const_inputs == other.const_inputs; in operator ==()
201 + s.const_inputs.size() * 2); in AbslHashValue()
221 absl::c_for_each(s.const_inputs, [&hashes](const ConstInput& const_input) { in AbslHashValue()
237 absl::flat_hash_set<string> const_inputs; member
465 absl::flat_hash_set<string>* const_inputs, in PushDownConstInputs() argument
486 const_inputs->insert(input); in PushDownConstInputs()
500 if (specialization.const_inputs.empty()) return; in RemovePushedDownConstInputs()
507 return !specialization.const_inputs.contains(input); in RemovePushedDownConstInputs()
535 if (specialization.const_inputs.empty()) return; in RemovePushedDownConstInputTypes()
[all …]
Dconstant_folding.cc3598 std::vector<int> const_inputs; in PartialAssocOpConstFolding() local
3605 const_inputs.push_back(i); in PartialAssocOpConstFolding()
3613 int const_inputs_size = const_inputs.size(); in PartialAssocOpConstFolding()
3633 for (int i : const_inputs) { in PartialAssocOpConstFolding()
3640 node->set_input(const_inputs[0], added_node->name()); in PartialAssocOpConstFolding()
3642 nonconst_inputs.push_back(const_inputs[0]); in PartialAssocOpConstFolding()
3653 const_inputs.size() - 1); in PartialAssocOpConstFolding()
3656 (*added_node->mutable_attr())["N"].set_i(const_inputs.size()); in PartialAssocOpConstFolding()
Dremapper.cc794 auto const_inputs = std::count_if( in IsCpuCompatibleDataType() local
800 bool can_remap = const_scaling_factor || const_inputs >= 4; in IsCpuCompatibleDataType()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dobject_reader.cc225 int const_inputs, in VerifyInputsConstsOutputs() argument
227 return CheckInputsConstsOutputs(context_, node, runtime_inputs, const_inputs, in VerifyInputsConstsOutputs()
Dmodel_builder_helper.cc238 int runtime_inputs, int const_inputs, in CheckInputsConstsOutputs() argument
242 if (const_inputs_from_model != const_inputs) { in CheckInputsConstsOutputs()
244 "Expected ", const_inputs, " const input tensor(s), but node has ", in CheckInputsConstsOutputs()
Dobject_reader.h99 int runtime_inputs, int const_inputs,
Dmodel_builder_helper.h72 int runtime_inputs, int const_inputs,
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-with-tf2xla.mlir130 // CHECK-LABEL: func @const_inputs
132 func @const_inputs(%arg0: tensor<2x2xf64>, %arg1: tensor<f64>, %arg2: tensor<2xi32>, %arg3: tensor<…