Searched refs:num_output_elements (Results 1 – 2 of 2) sorted by relevance
42 int num_output_elements = 1; in ResizeOutput() local50 num_output_elements *= value; in ResizeOutput()54 output_shape->data[stretch_dim] = num_input_elements / num_output_elements; in ResizeOutput()55 num_output_elements *= output_shape->data[stretch_dim]; in ResizeOutput()58 TF_LITE_ENSURE_EQ(context, num_input_elements, num_output_elements); in ResizeOutput()
479 const int64 num_output_elements = context->Value(num_elements_dim); in RegisterConstantNode() local480 data_size = max_bytes_per_data * num_output_elements; in RegisterConstantNode()929 const int64 num_output_elements = context->Value(num_elements_dim); in AppendNodeOutputParams() local930 data_size = max_bytes_per_data * num_output_elements; in AppendNodeOutputParams()933 CHECK_EQ(shapes.at(i).num_elements(), num_output_elements); in AppendNodeOutputParams()