Home
last modified time | relevance | path

Searched refs:AssignSubVariableOp (Results 1 – 24 of 24) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/
H A Dkeras_imagenet_main_graph_mode.pbtxt103 s: "loc:@bn2a_branch1_1/AssignMovingAvg/AssignSubVariableOp"
144 s: "loc:@bn2a_branch1_1/AssignMovingAvg_1/AssignSubVariableOp"
185 s: "loc:@bn2a_branch2a_1/AssignMovingAvg/AssignSubVariableOp"
226 s: "loc:@bn2a_branch2a_1/AssignMovingAvg_1/AssignSubVariableOp"
267 s: "loc:@bn2a_branch2b_1/AssignMovingAvg/AssignSubVariableOp"
308 s: "loc:@bn2a_branch2b_1/AssignMovingAvg_1/AssignSubVariableOp"
349 s: "loc:@bn2a_branch2c_1/AssignMovingAvg/AssignSubVariableOp"
390 s: "loc:@bn2a_branch2c_1/AssignMovingAvg_1/AssignSubVariableOp"
431 s: "loc:@bn2b_branch2a_1/AssignMovingAvg/AssignSubVariableOp"
472 s: "loc:@bn2b_branch2a_1/AssignMovingAvg_1/AssignSubVariableOp"
[all …]
H A Dkeras_imagenet_main.pbtxt34999 name: "bn_conv1_1/AssignMovingAvg_1/AssignSubVariableOp"
35000 op: "AssignSubVariableOp"
35012 name: "bn_conv1_1/AssignMovingAvg/AssignSubVariableOp"
35013 op: "AssignSubVariableOp"
35563 name: "bn2a_branch1_1/AssignMovingAvg_1/AssignSubVariableOp"
35564 op: "AssignSubVariableOp"
35576 name: "bn2a_branch1_1/AssignMovingAvg/AssignSubVariableOp"
35577 op: "AssignSubVariableOp"
35629 name: "bn2a_branch2a_1/AssignMovingAvg_1/AssignSubVariableOp"
35630 op: "AssignSubVariableOp"
[all …]
H A Dkeras_imagenet_main_graph_mode.golden_summary19 AssignSubVariableOp 106
H A Dkeras_imagenet_main.golden_summary52 AssignSubVariableOp 106
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Dresource_spmd_expander.cc149 mlir::TF::AssignSubVariableOp>(op)) in ExpandOp()
235 mlir::TF::AssignSubVariableOp>(op)) { in ComputeLayoutForward()
253 mlir::TF::AssignSubVariableOp>(op)) { in ComputeLayoutBackward()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dvariable_ops.cc123 class AssignSubVariableOp : public XlaOpKernel { class
125 explicit AssignSubVariableOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in AssignSubVariableOp() function in tensorflow::__anon735ed9830111::AssignSubVariableOp
138 AssignSubVariableOp);
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/java_api/
H A Dapi_def_AssignSubVariableOp.pbtxt2 graph_op_name: "AssignSubVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/python_api/
H A Dapi_def_AssignSubVariableOp.pbtxt2 graph_op_name: "AssignSubVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
H A DAssignSubVariableOp.pbtxt2 name: "AssignSubVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DAssignSubVariableOp.pbtxt2 name: "AssignSubVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_AssignSubVariableOp.pbtxt2 graph_op_name: "AssignSubVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expanders.cc237 REGISTER_SPMD(AssignSubVariable, TF::AssignSubVariableOp, ResourceSPMDExpander);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
H A Dtpu_cluster_formation.mlir353 // CHECK-NEXT: "tf.AssignSubVariableOp"([[UNUSED_RESOURCE]], [[CONST]])
362 …"tf.AssignSubVariableOp"(%arg1, %0) : (tensor<*x!tf_type.resource<tensor<f32>>>, tensor<f32>) -> ()
H A Ddecompose_resource_ops.mlir115 // Tests that composite tf.AssignSubVariableOp operation is decomposed using
130 …"tf.AssignSubVariableOp"(%0, %1) {dtype = "tfdtype$DT_INT32"} : (tensor<!tf_type.resource<tensor<i…
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/g3doc/
H A Dgpu_supported_ops.md22 `AssignSubVariableOp` | `dtype={complex64,double,float,int32,int64,uint32,uint64}`
H A Dcpu_supported_ops.md22 `AssignSubVariableOp` | `dtype={complex64,double,float,int32,int64,uint32,uint64}`
/aosp_15_r20/external/tensorflow/tensorflow/lite/g3doc/guide/
H A Dop_select_allowlist.md47 * `raw_ops.AssignSubVariableOp`
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/
H A Dlegalization.md1445 ### tf.AssignSubVariableOp
1452 %output = tf.AssignSubVariableOp(%resource, %value, %dtype)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.raw_ops.pbtxt268 name: "AssignSubVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.raw_ops.pbtxt268 name: "AssignSubVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/go/op/
H A Dwrappers.go1598 func AssignSubVariableOp(scope *Scope, resource tf.Output, value tf.Output) (o *tf.Operation) { func
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt2918 name: "AssignSubVariableOp"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
H A Dtf_generated_ops.td796 def TF_AssignSubVariableOp : TF_Op<"AssignSubVariableOp", []> {
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/
H A Dops_history.v2.pbtxt7343 name: "AssignSubVariableOp"