Home
last modified time | relevance | path

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

/external/executorch/backends/vulkan/runtime/
DVulkanBackend.cpp451 executorch::aten::SizesType new_output_size[kTensorDimensionLimit]; in maybe_resize_output() local
454 new_output_size[i] = out_tensor->sizes()[i]; in maybe_resize_output()
458 new_output_size, ndim}; in maybe_resize_output()
/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()