Home
last modified time | relevance | path

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

123

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
H A 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 …]
H A Dremote_run_encapsulate.cc190 auto new_op = builder.create<tfrt::dist::GetRemoteObjectIdFromTHOp>( in runOnOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/eliminate_passthrough_iter_args/
H A 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A 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
H A Dtensor_list_ops_decomposition.cc255 auto new_op = OpBuilder(op).create<CaseOrIfOp>( in HandleCaseOrIfOp() local
357 auto new_op = OpBuilder(if_op).create<TF::IfRegionOp>( in HandleIfRegionOp() local
402 auto new_op = OpBuilder(case_op).create<TF::CaseRegionOp>( in HandleCaseRegionOp() local
H A Drewrite_tpu_embedding_ops.cc42 auto new_op = builder->create<OpT>(op->getLoc(), op->getResultTypes(), in AddOperandAndRewriteAs() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A 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
/aosp_15_r20/system/update_engine/payload_generator/
H A Dab_generator.cc120 InstallOperation new_op; in SplitSourceCopy() local
174 InstallOperation new_op; in SplitAReplaceOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A 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
H A Dconvert_to_signless_pass.cc53 auto* new_op = Operation::create(op->getLoc(), op->getName(), result_types, in matchAndRewrite() local
/aosp_15_r20/system/security/keystore2/src/
H A Doperation.rs514 let new_op = Arc::new(Operation::new( in create_operation() localVariable
526 let new_op = Arc::new(Operation::new( in create_operation() localVariable
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/sparse_expansions/
H A Dmatmul_sparse_expander.cc36 mlir::TF::SparseTensorDenseMatMulOp new_op = in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/transforms/constant_folding/
H A Dpass.cc2383 Operation *new_op = rewriter.create(new_op_state); in matchAndRewrite() local
2601 Operation *new_op = rewriter.create(state); in matchAndRewrite() local
2678 Operation *new_op = rewriter.create(state); in matchAndRewrite() local
2765 Operation *new_op = rewriter.create(new_op_state); in matchAndRewrite() local
3104 Operation *new_op = rewriter.create(state); in matchAndRewrite() local
3131 Operation *new_op = rewriter.create(state); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/
H A Dnode_expansion_pass.cc79 auto* new_op = new EagerOperation(&ctx); in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dtensorlist_reserve_spmd_expander.cc50 mlir::TF::TensorListReserveOp new_op = in ExpandOp() local
H A Dtop_k_spmd_expander.cc66 mlir::Operation* new_op = builder.clone(*op, mapping); in ExpandOp() local
H A Din_top_k_spmd_expander.cc128 mlir::Operation* new_op = builder.clone(*op, mapping); in ExpandOp() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dall_reduce_reassociate.cc80 HloInstruction *new_op = computation->AddInstruction( in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
H A Drewrite_quantized_io.cc90 auto new_op = builder.create<TFR::CastOp>(returned_op->getLoc(), in runOnOperation() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Dlegalize_tf_while.cc63 auto new_op = OpBuilder(op).create<TFL::WhileOp>( in RunOnWhile() local
/aosp_15_r20/system/core/fs_mgr/libsnapshot/libsnapshot_cow/
H A Dparser_v2.cpp207 auto& new_op = out->ops->at(i); in Translate() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
H A Dconv_emitter_transforms.cc134 auto new_op = CloneWithNewAffineMap(owner, affine_map, OpBuilder(owner)); in TileLoop() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dfusion_bitcast_lift.cc172 HloInstruction* new_op = in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Ddevice_transform.cc119 Operation* new_op = builder->create(state); in ConvertQuantizedOpToFloat() local

123