Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dshape_inference.cc191 Status InferenceContext::ExpandOutputs(int new_output_size) { in ExpandOutputs() argument
193 if (new_output_size < outputs_size) { in ExpandOutputs()
196 outputs_.resize(new_output_size, nullptr); in ExpandOutputs()
197 output_handle_shapes_and_types_.resize(new_output_size); in ExpandOutputs()
Dshape_inference.h725 Status ExpandOutputs(int new_output_size);
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1624 int64_t new_output_size = in CalculateOutputSize() local
1626 if (new_output_size < 0) { in CalculateOutputSize()
1631 output_size = new_output_size; in CalculateOutputSize()