Home
last modified time | relevance | path

Searched defs:out_ty (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/csrc/jit/backends/
Dbackend_detail.cpp333 auto out_ty = schema.returns().at(0).type(); in codegen_backend_module() local
/external/pytorch/torch/csrc/jit/mobile/
Dpromoted_prim_ops.cpp157 at::TypePtr out_ty; in toList() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_a_m.cc2199 if (RankedTensorType out_ty = op.getType().dyn_cast<RankedTensorType>()) { in verify() local
2272 auto out_ty = op.getType().cast<TensorType>(); in verify() local
3115 Type out_ty = InferReductionOpType(input, reduction_indices, keep_dims); in build() local
Dtf_ops_n_z.cc1682 Type out_ty = InferReductionOpType(input, reduction_indices, keep_dims); in build() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlower_tf.cc293 RankedTensorType out_ty = in matchAndRewrite() local