| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms_test.cc | 47 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
|
| D | fuse_convolutions_test.cc | 52 Output input_op = in TestFuseResizePadAndConv() local
109 Output input_op = in TestFuseResizeAndConv() local
158 Output input_op = in TestFusePadAndConv() local
|
| D | fold_old_batch_norms_test.cc | 48 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
|
| D | round_weights_test.cc | 46 Output input_op = in TestRoundWeights() local
|
| D | flatten_atrous_test.cc | 47 Output input_op = in TestFlattenAtrousConv() local
|
| D | quantize_weights_test.cc | 47 Output input_op = in BuildGraphDef() local
|
| D | quantize_nodes_test.cc | 326 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
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | cost_model.cc | 60 Operation* input_op = input.getDefiningOp(); in GetTransferredTensorBytes() local
80 Operation* input_op = input.getDefiningOp(); in GetTransferredElementCount() local
|
| D | raise_target_subgraphs.cc | 140 Operation* input_op = input.getDefiningOp(); in CollectInputs() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | arg_op.cc | 70 auto input_op = arg.AsXlaOp(builder); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | split_merged_operands.cc | 91 Operation* input_op = operand.getDefiningOp(); in DuplicateValueIfNeeded() local
|
| D | post_quantize.cc | 162 auto input_op = op.input().getDefiningOp(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| D | post_quantize.cc | 76 auto input_op = op.getArg().getDefiningOp(); in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | handle_cross_cluster_dependencies.cc | 74 mlir::Operation* input_op = layout.input().getDefiningOp(); in GetConstOp() local
298 mlir::Operation* input_op = operand->getOwner(); in ReplaceCopyToMeshWithVirtualSendRecv() local
|
| D | tpu_add_resource_device_attribute.cc | 102 mlir::Operation* input_op = tpu_input.getDefiningOp(); in runOnOperation() local
|
| D | shape_utils.cc | 171 while (auto input_op = llvm::dyn_cast_or_null<mlir::TF::DTensorLayout>( in InferShapeOfTFOpWithCustomOperandConstantFn() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_dynamic_layout_pass.cc | 176 auto input_op = entry.value().get().getDefiningOp(); in HandleReplicatedInputs() local
242 auto* input_op = input.getDefiningOp(); in HandleCompileAndExecutes() local
|
| D | tpu_space_to_depth_pass.cc | 381 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
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_concatenation.cc | 155 const Operator* input_op = GetOpWithOutput(*model, input_name); in Run() local
|
| D | group_bidirectional_sequence_ops.cc | 87 Operator** input_op) { in FindUnidirectionalSequenceOp()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_util.py | 262 def CheckInputFromValidContext(op, input_op): argument
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 234 auto input_op = in VerifyFusedBatchNorm() local
359 auto input_op = in VerifyFusedBatchNormGradWithConv2D() local
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | operator.cpp | 428 std::vector<Symbol> findSimilarOperators(Symbol input_op) { in findSimilarOperators()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | op_builder.cc | 272 auto* input_op = AddNode(); in AddInputTensors() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 248 Operation* input_op = input.getDefiningOp(); in CollectInputsAndAttributes() local
|