Searched refs:const_input (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 238 absl::c_for_each(s.const_inputs, [&hashes](const ConstInput& const_input) { in AbslHashValue() argument
239 hashes.push_back(hash<InputPort>()(const_input.first)); in AbslHashValue()
240 hashes.push_back(Hash64(const_input.second)); in AbslHashValue()
556 const auto record_control_deps = [&](const NodeDef* const_input) { in PushDownConstInputs() argument
557 for (int i = const_input->input_size() - 1; i >= 0; --i) { in PushDownConstInputs()
558 const string& input = const_input->input(i); in PushDownConstInputs()
573 const auto* const_input = CHECK_NOTNULL(ctx.TrulyConstNode(node_name)); in PushDownConstInputs() local
575 record_control_deps(const_input); in PushDownConstInputs()
576 TF_RETURN_IF_ERROR(ReplaceInputWithConst(*const_input, i, item)); in PushDownConstInputs()
|
| D | constant_folding.cc | 2774 const string& const_input = node->input(1); in ReduceDivToReciprocalMul() local
2775 const NodeDef* denom = node_map_->GetNode(const_input); in ReduceDivToReciprocalMul()
2795 reciprocal_node->add_input(const_input); in ReduceDivToReciprocalMul()
2799 node_map_->UpdateOutput(node->name(), const_input, reciprocal_node->name()); in ReduceDivToReciprocalMul()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 1080 auto const_input = ops::Const(root, {9, 10, 11}); in TEST_F() local
1085 const_input, in TEST_F()
1164 auto const_input = ops::Const(root, {9, -2, 11}); in TEST_F() local
1169 const_input, in TEST_F()
|