Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/ops/
Dcontrol_flow_util.py266 def CheckInputFromValidContext(op, input_op): argument
281 input_ctxt = GetOutputContext(input_op)
347 " loops." % (input_op.name, op.name))
351 % (input_op.name, op.name, input_op.name))
358 log_msg += "\n%s while context: %s" % (input_op.name, input_while_ctxt)
361 input_op.name, "".join(traceback.format_list(input_op.traceback)))
Dclip_ops_test.py37 input_op = constant_op.constant(inputs)
38 clipped = clip_ops.clip_by_norm(input_op, max_norm)
Ddequantize_op_test.py37 input_op = constant_op.constant(inputs, shape=[len(inputs)], dtype=dtype)
38 dequantized = array_ops.dequantize(input_op, min_range, max_range,
Dcontrol_flow_grad.py98 input_op = op.inputs[0].op
101 op_ctxt = control_flow_util.GetOutputContext(input_op)
Dcritical_section_ops.py344 input_op for op in created_ops for input_op in op.control_inputs)
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dsplit_merged_operands.cc86 Operation* input_op = operand.getDefiningOp(); in DuplicateValueIfNeeded() local
87 if (input_op == nullptr) return failure(); in DuplicateValueIfNeeded()
90 if (!matchPattern(input_op, m_Constant(&attr))) { in DuplicateValueIfNeeded()
96 Operation* duplicated_input_op = builder->clone(*input_op); in DuplicateValueIfNeeded()
Dpost_quantize.cc131 auto input_op = op.input().getDefiningOp(); in matchAndRewrite() local
132 if (auto q = llvm::dyn_cast_or_null<QuantizeOp>(input_op)) { in matchAndRewrite()
/external/tensorflow/tensorflow/lite/toco/tflite/
Dimport.cc110 for (const auto* input_op : *ops) { in ImportOperators()
111 uint32_t index = input_op->opcode_index(); in ImportOperators()
126 ->Deserialize(input_op->builtin_options(), in ImportOperators()
127 input_op->custom_options()); in ImportOperators()
139 new_op = ops_by_name.at(opname)->Deserialize(input_op->builtin_options(), in ImportOperators()
140 input_op->custom_options()); in ImportOperators()
146 auto inputs = input_op->inputs(); in ImportOperators()
160 auto outputs = input_op->outputs(); in ImportOperators()
/external/tensorflow/tensorflow/python/data/util/
Dtraverse.py53 input_op = i.op
54 if input_op not in visited:
55 bfs_q.put(input_op)
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms_test.cc47 Output input_op = in TestFoldBatchNormsConv2D() local
56 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldBatchNormsConv2D()
98 Output input_op = in TestFoldBatchNormsDepthwiseConv2dNative() local
107 Output conv_op = DepthwiseConv2dNative(root.WithOpName("conv_op"), input_op, in TestFoldBatchNormsDepthwiseConv2dNative()
149 Output input_op = in TestFoldBatchNormsConv2DShared() local
158 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldBatchNormsConv2DShared()
207 Output input_op = in TestFoldBatchNormsMatMul() local
216 MatMul(root.WithOpName("matmul_op"), input_op, weights_op); in TestFoldBatchNormsMatMul()
Dfuse_convolutions_test.cc52 Output input_op = in TestFuseResizePadAndConv() local
55 Output resize_op = ResizeBilinear(root.WithOpName("resize_op"), input_op, in TestFuseResizePadAndConv()
109 Output input_op = in TestFuseResizeAndConv() local
112 Output resize_op = ResizeBilinear(root.WithOpName("resize_op"), input_op, in TestFuseResizeAndConv()
158 Output input_op = in TestFusePadAndConv() local
166 MirrorPad(root.WithOpName("pad_op"), input_op, pad_dims_op, "REFLECT"); in TestFusePadAndConv()
Dfold_old_batch_norms_test.cc48 Output input_op = in TestFoldOldBatchNorms() local
57 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldOldBatchNorms()
132 Output input_op = in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
141 Output conv_op = DepthwiseConv2dNative(root.WithOpName("conv_op"), input_op, in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
210 Output input_op = in TestFoldFusedBatchNorms() local
219 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldFusedBatchNorms()
287 Output input_op = in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
296 Output conv_op = DepthwiseConv2dNative(root.WithOpName("conv_op"), input_op, in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative()
466 Output input_op = in TestFoldFusedBatchNormsWithBatchToSpace() local
475 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, weights_op, in TestFoldFusedBatchNormsWithBatchToSpace()
Dround_weights_test.cc46 Output input_op = in TestRoundWeights() local
59 Output conv_op = Conv2D(root.WithOpName("output"), input_op, weights_op, in TestRoundWeights()
Dflatten_atrous_test.cc47 Output input_op = in TestFlattenAtrousConv() local
61 SpaceToBatchND(root.WithOpName("space_to_batch_op"), input_op, in TestFlattenAtrousConv()
Dquantize_weights_test.cc47 Output input_op = in BuildGraphDef() local
55 Output conv_op = ops::Conv2D(root.WithOpName("output"), input_op, in BuildGraphDef()
Dquantize_nodes_test.cc326 Output input_op = in TestQuantizeConv2D() local
335 Output conv_op = Conv2D(root.WithOpName("conv_op"), input_op, filter_op, in TestQuantizeConv2D()
350 Output input_op = in TestQuantizeBiasAdd() local
359 BiasAdd(root.WithOpName("bias_add_op"), input_op, offset_op); in TestQuantizeBiasAdd()
881 Output input_op = in TestConvertFakeQuantsToRequantize() local
890 BiasAdd(root.WithOpName("bias_add_op"), input_op, offset_op); in TestConvertFakeQuantsToRequantize()
926 Output input_op = in TestMergeAdjacentRequantizes() local
955 root.WithOpName("quantized_bias_add_op"), input_op, offset_op, in TestMergeAdjacentRequantizes()
1033 Output input_op = in TestConvertFakeQuantsEndToEnd() local
1042 BiasAdd(root.WithOpName("bias_add_op"), input_op, offset_op); in TestConvertFakeQuantsEndToEnd()
[all …]
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_test.py135 input_op = constant_op.constant(
141 input_op, weights_op, [1, 1, 1, 1], padding="SAME", name="conv_op")
184 input_op = constant_op.constant(
197 input_op,
244 input_op = constant_op.constant(
266 input_op,
300 input_op = constant_op.constant(
303 input_op, [12, 4], align_corners=False)
332 input_op = constant_op.constant(
335 input_op, [12, 4], align_corners=False)
[all …]
Doptimize_for_inference_lib.py461 input_op = node_from_map(input_node_map, conv_op.input[0])
462 if input_op.op == "MirrorPad":
463 mirror_pad_op = input_op
469 if input_op.op == "ResizeBilinear":
470 resize_op = input_op
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Darg_op.cc69 auto input_op = arg.AsXlaOp(builder); in Compile() local
72 0, xla::CustomCall(builder, "SetBound", {input_op}, in Compile()
73 builder->GetShape(input_op).ValueOrDie(), "", in Compile()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_dynamic_layout_pass.cc193 auto input_op = entry.value().get().getDefiningOp(); in HandleReplicatedInputs() local
194 if (!input_op || !IsSupportedInputOp(input_op, resource_alias_analysis)) in HandleReplicatedInputs()
259 auto* input_op = input.getDefiningOp(); in HandleCompileAndExecutes() local
261 maybe_replicate.body().isAncestor(input_op->getParentRegion())) { in HandleCompileAndExecutes()
264 if (!IsSupportedInputOp(input_op, resource_alias_analysis)) continue; in HandleCompileAndExecutes()
Dtpu_space_to_depth_pass.cc426 auto input_op = input.getDefiningOp(); in BuildSpaceToDepth() local
427 OpBuilder builder(input_op); in BuildSpaceToDepth()
428 builder.setInsertionPointAfter(input_op); in BuildSpaceToDepth()
462 auto input_op = input.get().getDefiningOp(); in HandleHostReplicatedInputs() local
463 if (!input_op || !IsSupportedHostInputOp(input_op)) return false; in HandleHostReplicatedInputs()
501 auto input_op = input.value().getDefiningOp(); in HandleCluster() local
503 maybe_replicate.body().isAncestor(input_op->getParentRegion())) { in HandleCluster()
506 if (!IsSupportedHostInputOp(input_op)) continue; in HandleCluster()
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dop_builder.cc272 auto* input_op = AddNode(); in AddInputTensors() local
273 input_op->SetOpType(OP_INPUT); in AddInputTensors()
282 input_op->AddOutput(tensor.dims, GetElementSize(tensor.type)); in AddInputTensors()
283 AddTensorWithID(tensor_id, input_op->GetID(), num_inputs); in AddInputTensors()
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_constant_concatenation.cc154 const Operator* input_op = GetOpWithOutput(*model, input_name); in Run() local
155 if (input_op) return ::tensorflow::Status::OK(); in Run()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op_test.cc233 auto input_op = in VerifyFusedBatchNorm() local
248 auto bn = ops::FusedBatchNorm(root.WithOpName("FusedBatchNorm"), input_op, in VerifyFusedBatchNorm()
356 auto input_op = in VerifyFusedBatchNormGradWithConv2D() local
362 auto conv = ops::Conv2D(root.WithOpName("Conv"), input_op, filter_op, in VerifyFusedBatchNormGradWithConv2D()
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc244 Operation* input_op = input.getDefiningOp(); in CollectInputsAndAttributes() local
251 if (auto cast_op = dyn_cast_or_null<CastOp>(input_op)) { in CollectInputsAndAttributes()
267 if (auto list_op = dyn_cast_or_null<BuildListOp>(input_op)) { in CollectInputsAndAttributes()

12