Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
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
Dfuse_convolutions_test.cc52 Output input_op = in TestFuseResizePadAndConv() local
109 Output input_op = in TestFuseResizeAndConv() local
158 Output input_op = in TestFusePadAndConv() local
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
Dround_weights_test.cc46 Output input_op = in TestRoundWeights() local
Dflatten_atrous_test.cc47 Output input_op = in TestFlattenAtrousConv() local
Dquantize_weights_test.cc47 Output input_op = in BuildGraphDef() local
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
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Darg_op.cc69 auto input_op = arg.AsXlaOp(builder); in Compile() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dsplit_merged_operands.cc86 Operation* input_op = operand.getDefiningOp(); in DuplicateValueIfNeeded() local
Dpost_quantize.cc131 auto input_op = op.input().getDefiningOp(); in matchAndRewrite() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_layout_pass.cc193 auto input_op = entry.value().get().getDefiningOp(); in HandleReplicatedInputs() local
259 auto* input_op = input.getDefiningOp(); in HandleCompileAndExecutes() local
Dtpu_space_to_depth_pass.cc426 auto input_op = input.getDefiningOp(); in BuildSpaceToDepth() local
462 auto input_op = input.get().getDefiningOp(); in HandleHostReplicatedInputs() local
501 auto input_op = input.value().getDefiningOp(); in HandleCluster() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_concatenation.cc154 const Operator* input_op = GetOpWithOutput(*model, input_name); in Run() local
Dgroup_bidirectional_sequence_ops.cc86 Operator** input_op) { in FindUnidirectionalSequenceOp()
/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util.py266 def CheckInputFromValidContext(op, input_op): argument
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op_test.cc233 auto input_op = in VerifyFusedBatchNorm() local
356 auto input_op = in VerifyFusedBatchNormGradWithConv2D() local
Dmkl_fused_ops_test.cc1066 auto input_op = in VerifyFusedMatMul() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dop_builder.cc272 auto* input_op = AddNode(); in AddInputTensors() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc244 Operation* input_op = input.getDefiningOp(); in CollectInputsAndAttributes() local
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_utils.cc614 auto input_op = q.arg().getDefiningOp(); in RemoveRedundantStatsOps() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc1600 TFL::ConstOp NarrowDownInt64InputValuesForOp(Operation *input_op, in NarrowDownInt64InputValuesForOp()
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1430 TF_Operation* input_op = in TEST() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc3376 const string input_op = in ConstantPushDown() local