Searched refs:ReplaceInputWithConst (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/utils/ |
D | functions.h | 185 friend Status ReplaceInputWithConst(const NodeDef&, int, 243 Status ReplaceInputWithConst(const NodeDef& input_const, int input_index,
|
D | functions_test.cc | 732 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()
|
D | functions.cc | 679 Status ReplaceInputWithConst(const NodeDef& input_const, int input_index, in ReplaceInputWithConst() function
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | graph_properties.cc | 666 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/ |
D | function_optimizer.cc | 576 TF_RETURN_IF_ERROR(ReplaceInputWithConst(*const_input, i, item)); in PushDownConstInputs()
|