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