Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h185 friend Status ReplaceInputWithConst(const NodeDef&, int,
243 Status ReplaceInputWithConst(const NodeDef& input_const, int input_index,
Dfunctions_test.cc732 TEST_F(FunctionsTest, ReplaceInputWithConst) { in TEST_F() argument
770 TF_EXPECT_OK(ReplaceInputWithConst(const_input_x, 0, &item)); in TEST_F()
777 TF_EXPECT_OK(ReplaceInputWithConst(const_input_y, 0, &item)); in TEST_F()
Dfunctions.cc679 Status ReplaceInputWithConst(const NodeDef& input_const, int input_index, in ReplaceInputWithConst() function
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc666 ReplaceInputWithConst(*input_node, i, &grappler_function_item)); in UpdateFunction()
671 TF_CHECK_OK(ReplaceInputWithConst(const_input_node, i, in UpdateFunction()
682 TF_CHECK_OK(ReplaceInputWithConst(const_input_node, i, in UpdateFunction()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dfunction_optimizer.cc576 TF_RETURN_IF_ERROR(ReplaceInputWithConst(*const_input, i, item)); in PushDownConstInputs()