Home
last modified time | relevance | path

Searched defs:output_ty (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_ops_n_z.cc344 auto output_ty = pack_op.output().getType().cast<ShapedType>(); in matchAndRewrite() local
749 TensorType &output_ty) { in GetReshapeOutputType()
848 auto output_ty = op.getType().dyn_cast<RankedTensorType>(); in verify() local
876 TensorType output_ty; in build() local
1267 auto output_ty = op.output().getType().dyn_cast<RankedTensorType>(); in verify() local
2128 auto output_ty = output().getType().dyn_cast<RankedTensorType>(); in fold() local
H A Dtf_ops_a_m.cc257 RankedTensorType output_ty = GetRankedTensorTypeForOperand(op.output()); in Verify() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Drewrite_tpu_embedding_ops.cc77 auto output_ty = in RunOnRegion() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_vectorize_convolutions.cc64 PrimitiveType output_ty = instr->shape().tuple_shapes(0).element_type(); in GetRelevantConvs() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_tfl.cc2501 ShapedType output_ty = in matchAndRewrite() local
2605 ShapedType output_ty = in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Doptimize.cc1365 auto output_ty = fully_connected_op.output()[0] in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc1757 TensorType &output_ty) { in GetReshapeOutputType()
1858 auto output_ty = op.getType().dyn_cast<RankedTensorType>(); in verify() local
/aosp_15_r20/external/rust/android-crates-io/crates/mockall_derive/src/
Dmock_function.rs232 let mut output_ty = supersuperfy(ty, self.levels); in build() localVariable