Home
last modified time | relevance | path

Searched defs:input_op (Results 1 – 25 of 36) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_batch_norms_test.cc47 Output input_op = in TestFoldBatchNormsConv2D() local
98 Output input_op = in TestFoldBatchNormsDepthwiseConv2dNative() local
149 Output input_op = in TestFoldBatchNormsConv2DShared() local
207 Output input_op = in TestFoldBatchNormsMatMul() local
H A Dfold_old_batch_norms_test.cc48 Output input_op = in TestFoldOldBatchNorms() local
132 Output input_op = in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
210 Output input_op = in TestFoldFusedBatchNorms() local
287 Output input_op = in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
466 Output input_op = in TestFoldFusedBatchNormsWithBatchToSpace() local
H A Dfuse_convolutions_test.cc52 Output input_op = in TestFuseResizePadAndConv() local
109 Output input_op = in TestFuseResizeAndConv() local
158 Output input_op = in TestFusePadAndConv() local
H A Dquantize_nodes_test.cc326 Output input_op = in TestQuantizeConv2D() local
350 Output input_op = in TestQuantizeBiasAdd() local
881 Output input_op = in TestConvertFakeQuantsToRequantize() local
926 Output input_op = in TestMergeAdjacentRequantizes() local
1033 Output input_op = in TestConvertFakeQuantsEndToEnd() local
1082 Output input_op = in TestHoistFakeQuants() local
H A Dround_weights_test.cc46 Output input_op = in TestRoundWeights() local
H A Dflatten_atrous_test.cc47 Output input_op = in TestFlattenAtrousConv() local
H A Dquantize_weights_test.cc47 Output input_op = in BuildGraphDef() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Dcost_model.cc60 Operation* input_op = input.getDefiningOp(); in GetTransferredTensorBytes() local
80 Operation* input_op = input.getDefiningOp(); in GetTransferredElementCount() local
H A Draise_target_subgraphs.cc140 Operation* input_op = input.getDefiningOp(); in CollectInputs() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dhandle_cross_cluster_dependencies.cc74 mlir::Operation* input_op = layout.input().getDefiningOp(); in GetConstOp() local
298 mlir::Operation* input_op = operand->getOwner(); in ReplaceCopyToMeshWithVirtualSendRecv() local
H A Dtpu_add_resource_device_attribute.cc102 mlir::Operation* input_op = tpu_input.getDefiningOp(); in runOnOperation() local
H A Dshape_utils.cc171 while (auto input_op = llvm::dyn_cast_or_null<mlir::TF::DTensorLayout>( in InferShapeOfTFOpWithCustomOperandConstantFn() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_dynamic_layout_pass.cc176 auto input_op = entry.value().get().getDefiningOp(); in HandleReplicatedInputs() local
242 auto* input_op = input.getDefiningOp(); in HandleCompileAndExecutes() local
H A Dtpu_space_to_depth_pass.cc381 auto input_op = input.getDefiningOp(); in BuildSpaceToDepth() local
417 auto input_op = input.get().getDefiningOp(); in HandleHostReplicatedInputs() local
456 auto input_op = input.value().getDefiningOp(); in HandleCluster() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_batch_norm_op_test.cc234 auto input_op = in VerifyFusedBatchNorm() local
359 auto input_op = in VerifyFusedBatchNormGradWithConv2D() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Darg_op.cc70 auto input_op = arg.AsXlaOp(builder); in Compile() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dsplit_merged_operands.cc91 Operation* input_op = operand.getDefiningOp(); in DuplicateValueIfNeeded() local
H A Dpost_quantize.cc162 auto input_op = op.input().getDefiningOp(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dpost_quantize.cc76 auto input_op = op.getArg().getDefiningOp(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dresolve_constant_concatenation.cc155 const Operator* input_op = GetOpWithOutput(*model, input_name); in Run() local
H A Dgroup_bidirectional_sequence_ops.cc87 Operator** input_op) { in FindUnidirectionalSequenceOp()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dcontrol_flow_util.py262 def CheckInputFromValidContext(op, input_op): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Doperator.cpp428 std::vector<Symbol> findSimilarOperators(Symbol input_op) { in findSimilarOperators()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dop_builder.cc272 auto* input_op = AddNode(); in AddInputTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
H A Draise_to_tf.cc248 Operation* input_op = input.getDefiningOp(); in CollectInputsAndAttributes() local

12