Home
last modified time | relevance | path

Searched defs:new_op (Results 1 – 25 of 63) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dconstant_op_device_assignment.cc65 mlir::Operation *new_op = builder.clone(*op.getOperation()); in runOnOperation() local
70 mlir::Operation *new_op = cloned_op_by_device[device_attr.getValue()]; in runOnOperation() local
Drewrite_tpu_embedding_ops.cc42 auto new_op = builder->create<OpT>(op->getLoc(), op->getResultTypes(), in AddOperandAndRewriteAs() local
/external/tensorflow/tensorflow/core/transforms/eliminate_passthrough_iter_args/
Dpass.cc104 OpT new_op = ConcreteT::RebuildOp(remove, op, rewriter); in Run() local
130 auto new_op = rewriter.create<ForRegionOp>( in RebuildOp() local
159 auto new_op = rewriter.create<WhileLikeRegionOp>( in RebuildOp() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dtf_to_tfrt.cc269 auto new_op = rewriter.create<tfrt::fallback_async::ExecuteOp>( in ConvertToFallbackExecuteOp() local
282 auto new_op = rewriter.create<tfrt::fallback_async::ExecuteOp>( in ConvertToFallbackExecuteOp() local
289 auto new_op = rewriter.create<tfrt::fallback_async::ExecuteOpSeq>( in ConvertToFallbackExecuteOp() local
326 auto new_op = rewriter.create<tfrt::corert::ExecuteOp>( in ConvertToCoreRTExecuteOp() local
332 auto new_op = rewriter.create<tfrt::corert::ExecuteOpSeq>( in ConvertToCoreRTExecuteOp() local
405 auto new_op = rewriter.create<tfrt::fallback_async::SetResourceOp>( in matchAndRewrite() local
443 auto new_op = rewriter.create<tfrt::fallback_async::GetResourceOp>( in matchAndRewrite() local
600 auto new_op = rewriter.create<tfrt::fallback_async::BatchFunctionOp>( in matchAndRewrite() local
637 auto new_op = rewriter.create<tfrt::corert::ConstDenseTensorOp>( in matchAndRewrite() local
770 auto new_op = rewriter.create<tfrt::corert::ConstStringTensorOp>( in matchAndRewrite() local
[all …]
Dremote_run_encapsulate.cc190 auto new_op = builder.create<tfrt::dist::GetRemoteObjectIdFromTHOp>( in runOnOperation() local
/external/tensorflow/tensorflow/dtensor/mlir/sparse_expansions/
Dmatmul_sparse_expander.cc36 mlir::TF::SparseTensorDenseMatMulOp new_op = in ExpandOp() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
Dnode_expansion_pass.cc79 auto* new_op = new EagerOperation(&ctx); in Run() local
/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dtensorlist_reserve_spmd_expander.cc50 mlir::TF::TensorListReserveOp new_op = in ExpandOp() local
Dtop_k_spmd_expander.cc66 mlir::Operation* new_op = builder.clone(*op, mapping); in ExpandOp() local
Din_top_k_spmd_expander.cc128 mlir::Operation* new_op = builder.clone(*op, mapping); in ExpandOp() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dall_reduce_reassociate.cc80 HloInstruction *new_op = computation->AddInstruction( in Run() local
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Drewrite_quantized_io.cc90 auto new_op = builder.create<TFR::CastOp>(returned_op->getLoc(), in runOnOperation() local
Ddecompose.cc235 auto new_op = builder.create<CallOp>( in RewriteUnregisteredTFOps() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf_while.cc63 auto new_op = OpBuilder(op).create<TFL::WhileOp>( in RunOnWhile() local
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dconvert_to_signless_pass.cc53 auto* new_op = Operation::create(op->getLoc(), op->getName(), result_types, in matchAndRewrite() local
Dtf_to_jit_invocations.cc162 auto new_op = rewriter.clone(*op, bvm); in matchAndRewrite() local
173 auto new_op = rewriter.clone(*op); in matchAndRewrite() local
/external/rust/android-crates-io/crates/spdx/src/expression/
Dparser.rs224 let new_op = match lt.token { in parse_mode() localVariable
/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter_transforms.cc134 auto new_op = CloneWithNewAffineMap(owner, affine_map, OpBuilder(owner)); in TileLoop() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfusion_bitcast_lift.cc172 HloInstruction* new_op = in Run() local
/external/tensorflow/tensorflow/core/framework/
Dop_def_util.cc623 Status OpDefCompatible(const OpDef& old_op, const OpDef& new_op) { in OpDefCompatible()
694 const OpDef& new_op) { in OpDefAddedDefaultsUnchanged()
731 Status OpDefAttrDefaultsUnchanged(const OpDef& old_op, const OpDef& new_op) { in OpDefAttrDefaultsUnchanged()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc1654 tensorflow::NodeDef* new_op = tensorflow_graph->add_node(); in ConvertSpaceToBatchNDOperator() local
1671 tensorflow::NodeDef* new_op = tensorflow_graph->add_node(); in ConvertBatchToSpaceNDOperator() local
1687 tensorflow::NodeDef* new_op = tensorflow_graph->add_node(); in ConvertPadOperator() local
1718 tensorflow::NodeDef* new_op = tensorflow_graph->add_node(); in ConvertPadV2Operator() local
1768 tensorflow::NodeDef* new_op = tensorflow_graph->add_node(); in ConvertStridedSliceOperator() local
1796 tensorflow::NodeDef* new_op = tensorflow_graph->add_node(); in ConvertSliceOperator() local
1818 tensorflow::NodeDef* new_op = tensorflow_graph->add_node(); in ConvertReduceOperator() local
1855 tensorflow::NodeDef* new_op = tensorflow_graph->add_node(); in ConvertSqueezeOperator() local
1963 tensorflow::NodeDef* new_op = tensorflow_graph->add_node(); in ConvertRandomUniformOperator() local
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform.cc119 Operation* new_op = builder->create(state); in ConvertQuantizedOpToFloat() local
/external/tensorflow/tensorflow/lite/toco/tflite/
Dimport.cc122 std::unique_ptr<Operator> new_op = nullptr; in ImportOperators() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_legalize_i1_type.cc177 Operation *new_op = rewriter.create(new_op_state); in matchAndRewrite() local
/external/pdfium/core/fpdfapi/page/
Dcpdf_psengine_unittest.cpp80 PDF_PSOP new_op = new_psop->GetOp(); in TEST() local

123