Searched refs:fused_op (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 1072 NodeDef fused_op; in IsCpuCompatibleDataType() local
1073 fused_op.set_name(bias_add.name()); in IsCpuCompatibleDataType()
1074 fused_op.set_device(contraction.device()); in IsCpuCompatibleDataType()
1075 fused_op.add_input(contraction.input(0)); // 0: input in IsCpuCompatibleDataType()
1076 fused_op.add_input(contraction.input(1)); // 1: filter in IsCpuCompatibleDataType()
1077 fused_op.add_input(bias_add.input(1)); // 2: bias in IsCpuCompatibleDataType()
1080 fused_op.set_op(kFusedConv2D); in IsCpuCompatibleDataType()
1081 CopyConv2DAttributes(contraction, &fused_op); in IsCpuCompatibleDataType()
1083 fused_op.set_op(kFusedDepthwiseConv2dNative); in IsCpuCompatibleDataType()
1084 CopyDepthwiseConv2dNativeAttributes(contraction, &fused_op); in IsCpuCompatibleDataType()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tf_data_optimization.cc | 44 auto fused_op = rewriter.create<MapAndBatchDatasetOp>( in matchAndRewrite() local
50 rewriter.replaceOp(op, {fused_op.handle()}); in matchAndRewrite()
|
| D | contraction_fusion.cc | 62 } else if (FusedOp fused_op = operand.getDefiningOp<FusedOp>()) { in matchAndRewrite() local
63 fuse_into = fused_op.getOperation(); in matchAndRewrite()
|
| D | fused_kernel_matcher.cc | 175 Value fused_op = rewriter.create<FusedOpT>(fused_loc, result_type, in matchAndRewrite() local
178 rewriter.replaceOp(op_to_replace, ValueRange({fused_op})); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tfrt_ops.td | 35 input and the output of each fused_op must be of type T.
|
| D | tf_generated_ops.td | 18337 the output of each fused_op must be of type T.
18339 Currently supported fused_op combinations are: [X] and [X,A], where X is one of
18386 the output of each fused_op must be of type T.
18388 Currently supported fused_op combinations are: ["BiasAdd"] and ["BiasAdd",A],
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 2064 for (auto& fused_op : fused_op_contexts) { in PredictFusedOp() local
2066 s.Update(PredictNodeCosts(fused_op, &fused_node_costs)); in PredictFusedOp()
|