Home
last modified time | relevance | path

Searched defs:var_op (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_initialized_variables.cc73 auto& var_op = std::get<0>(var_and_tensor); in MarkInitializedVariablesInFunction() local
Dinitialize_variables_in_session_init.cc161 auto& var_op = std::get<0>(var_and_tensor); in runOnOperation() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op_test.cc242 auto var_op = in VerifyFusedBatchNorm() local
374 auto var_op = ops::Const(root.WithOpName("variance"), in VerifyFusedBatchNormGradWithConv2D() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/
Dlower_saved_model.cc851 mlir::TF::VariableV2Op var_op) { in ConvertReferenceVariableToResourceVariable()
958 for (auto var_op : var_ops) { in runOnOperation() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc1780 TFE_Op* var_op = TFE_NewOp(ctx, "VarHandleOp", status); in TestOpAddAttrs() local
1836 TFE_Op* var_op = TFE_NewOp(ctx, "VarHandleOp", status); in TEST() local