Searched refs:var_op (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | legalize_variables.cc | 97 T var_op, ArrayRef<Value> operands, in matchAndRewrite() argument
99 auto* op = var_op.getOperation(); in matchAndRewrite()
100 auto func = var_op->template getParentOfType<FuncOp>(); in matchAndRewrite()
102 auto global_tensor = GetGlobalTensor<T>(symbol_table_, var_op, func); in matchAndRewrite()
105 mlir::OpBuilder builder(var_op); in matchAndRewrite()
114 var_op->replaceAllUsesWith(tfl_var_op); in matchAndRewrite()
115 rewriter.eraseOp(var_op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_lib.py | 299 var_op = node_from_map(input_node_map,
301 if var_op.op != "Const":
304 " run first?" % (node.name, var_op))
306 var_value = values_from_const(var_op)
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 241 auto var_op = in VerifyFusedBatchNorm() local
249 scale_op, offset_op, mean_op, var_op, attr); in VerifyFusedBatchNorm()
371 auto var_op = ops::Const(root.WithOpName("variance"), in VerifyFusedBatchNormGradWithConv2D() local
381 scale_op, mean_op, var_op, res_sp3_op, bn_attr); in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 1673 TFE_Op* var_op = TFE_NewOp(ctx, "VarHandleOp", status); in TEST() local
1674 TFE_OpSetAttrType(var_op, "dtype", TF_INT64); in TEST()
1675 TFE_OpSetAttrShape(var_op, "shape", {}, 0, status); in TEST()
1676 const TFE_OpAttrs* attributes = TFE_OpGetAttrs(var_op); in TEST()
1696 TFE_DeleteOp(var_op); in TEST()
1708 TFE_Op* var_op = TFE_NewOp(ctx, "VarHandleOp", status); in TEST() local
1710 TFE_OpSetAttrType(var_op, "dtype", TF_INT64); in TEST()
1711 TFE_OpSetAttrShape(var_op, "shape", {}, 0, status); in TEST()
1712 const TFE_OpAttrs* attributes = TFE_OpGetAttrs(var_op); in TEST()
1743 TFE_DeleteOp(var_op); in TEST()
|