Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc609 GrapplerFunctionItem grappler_function_item = it->second; in UpdateFunction() local
610 MutableGraphView gv(&grappler_function_item.graph); in UpdateFunction()
613 for (int i = 0; i < grappler_function_item.inputs().size(); ++i) { in UpdateFunction()
614 auto& fun_input = grappler_function_item.input(i); in UpdateFunction()
660 for (int i = grappler_function_item.inputs().size() - 1; i >= 0; --i) { in UpdateFunction()
666 ReplaceInputWithConst(*input_node, i, &grappler_function_item)); in UpdateFunction()
672 &grappler_function_item)); in UpdateFunction()
683 &grappler_function_item)); in UpdateFunction()
688 GraphProperties gp(grappler_function_item); in UpdateFunction()
693 ctx->output_tensors_as_shapes.resize(grappler_function_item.output_size()); in UpdateFunction()
[all …]