Searched refs:const_input (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 221 absl::c_for_each(s.const_inputs, [&hashes](const ConstInput& const_input) { in AbslHashValue() argument
222 hashes.push_back(hash<InputPort>()(const_input.first)); in AbslHashValue()
223 hashes.push_back(Hash64(const_input.second)); in AbslHashValue()
468 const auto record_control_deps = [&](const NodeDef* const_input) { in PushDownConstInputs() argument
469 for (int i = const_input->input_size() - 1; i >= 0; --i) { in PushDownConstInputs()
470 const string& input = const_input->input(i); in PushDownConstInputs()
485 const auto* const_input = CHECK_NOTNULL(ctx.TrulyConstNode(node_name)); in PushDownConstInputs() local
487 record_control_deps(const_input); in PushDownConstInputs()
488 TF_RETURN_IF_ERROR(ReplaceInputWithConst(*const_input, i, item)); in PushDownConstInputs()
|
| D | constant_folding.cc | 3052 const string& const_input = node->input(1); in ReduceDivToReciprocalMul() local
3053 const NodeDef* denom = node_map_->GetNode(const_input); in ReduceDivToReciprocalMul()
3072 reciprocal_node->add_input(const_input); in ReduceDivToReciprocalMul()
3085 node_map_->UpdateOutput(node->name(), const_input, reciprocal_node->name()); in ReduceDivToReciprocalMul()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | mark_for_compilation_pass.cc | 699 Node* const_input = nullptr; in IsScalarIntegerResourceOperation() local
702 const_input = e->src(); in IsScalarIntegerResourceOperation()
707 if (!const_input) { in IsScalarIntegerResourceOperation()
712 if (!TryGetNodeAttr(const_input->def(), "value", &proto)) { in IsScalarIntegerResourceOperation()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner_test.cc | 1065 auto const_input = ops::Const(root, {9, 10, 11}); in TEST_F() local
1070 const_input, in TEST_F()
1149 auto const_input = ops::Const(root, {9, -2, 11}); in TEST_F() local
1154 const_input, in TEST_F()
|