Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h91 friend Status ReplaceInputWithConst(const NodeDef&, int,
146 Status ReplaceInputWithConst(const NodeDef& input_const, int input_index,
Dfunctions_test.cc576 TEST_F(FunctionsTest, ReplaceInputWithConst) { in TEST_F() argument
614 TF_EXPECT_OK(ReplaceInputWithConst(const_input_x, 0, &item)); in TEST_F()
621 TF_EXPECT_OK(ReplaceInputWithConst(const_input_y, 0, &item)); in TEST_F()
Dfunctions.cc310 Status ReplaceInputWithConst(const NodeDef& input_const, int input_index, in ReplaceInputWithConst() function
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc876 ReplaceInputWithConst(*input_node, i, &grappler_function_item)); in UpdateFunction()
881 TF_CHECK_OK(ReplaceInputWithConst(const_input_node, i, in UpdateFunction()
892 TF_CHECK_OK(ReplaceInputWithConst(const_input_node, i, in UpdateFunction()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc488 TF_RETURN_IF_ERROR(ReplaceInputWithConst(*const_input, i, item)); in PushDownConstInputs()