Searched defs:input_tensor_type (Results 1 – 10 of 10) sorted by relevance
313 auto input_tensor_type = input->type()->cast<TensorType>(); in UpdateScalarTypeForInputs() local407 auto input_tensor_type = in LowPrecisionCastNodeForStandardOps() local
2239 auto input_tensor_type = graph->inputs()[i]->type()->cast<TensorType>(); in ONNXSetDynamicInputShape() local
185 const ::tflite::TensorType input_tensor_type = in CreateTfLiteModel() local
156 auto input_tensor_type = input->type()->cast<TensorType>(); in castTensorInputs() local
194 TensorTypePtr input_tensor_type = nullptr; in removeProfileNodesAndSpecializeTypes() local
114 auto input_tensor_type = in PadInputOnUnshardedDim() local
450 auto input_tensor_type = tensor.getType().dyn_cast<mlir::RankedTensorType>(); in EmitHaloExchange() local
54 auto input_tensor_type = inp->type()->cast<c10::TensorType>(); in moveCatAfterUse() local
539 auto input_tensor_type = in HandleFirstConvolution() local
454 auto input_tensor_type = input_type.dyn_cast<TensorType>(); in matchAndRewrite() local