Lines Matching refs:subgraph_output_type

148       auto* subgraph_output_type = output_types[i];  in ScanInferenceFunction()  local
153       if (!subgraph_output_type->has_tensor_type()) {  in ScanInferenceFunction()
159       auto& subgraph_output_tensor_type = subgraph_output_type->tensor_type();  in ScanInferenceFunction()
328       auto* subgraph_output_type = subgraph_output_types[i + 1]; // skip 'cond'  in LoopInferenceFunction()  local
333       if (!subgraph_output_type->has_tensor_type() && !subgraph_output_type->has_sequence_type()) {  in LoopInferenceFunction()
338             subgraph_output_type->value_case());  in LoopInferenceFunction()
341       if (!is_loop_state_var && !subgraph_output_type->has_tensor_type()) {  in LoopInferenceFunction()
346             subgraph_output_type->value_case());  in LoopInferenceFunction()
350       propagateElemTypeWithValidation(subgraph_output_type, loop_output_type);  in LoopInferenceFunction()
356         if (subgraph_output_type->tensor_type().has_shape()) {  in LoopInferenceFunction()
359           TypeProto inferred_type(*subgraph_output_type);  in LoopInferenceFunction()
372                subgraph_output_type->tensor_type().shape().dim()) {  in LoopInferenceFunction()