/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/tests/ |
H A D | keras_imagenet_main_graph_mode.pbtxt | 103 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 D | keras_imagenet_main.pbtxt | 34999 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 D | keras_imagenet_main_graph_mode.golden_summary | 19 AssignSubVariableOp 106
|
H A D | keras_imagenet_main.golden_summary | 52 AssignSubVariableOp 106
|
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
H A D | resource_spmd_expander.cc | 149 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 D | variable_ops.cc | 123 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 D | api_def_AssignSubVariableOp.pbtxt | 2 graph_op_name: "AssignSubVariableOp"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/python_api/ |
H A D | api_def_AssignSubVariableOp.pbtxt | 2 graph_op_name: "AssignSubVariableOp"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
H A D | AssignSubVariableOp.pbtxt | 2 name: "AssignSubVariableOp"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
H A D | AssignSubVariableOp.pbtxt | 2 name: "AssignSubVariableOp"
|
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
H A D | api_def_AssignSubVariableOp.pbtxt | 2 graph_op_name: "AssignSubVariableOp"
|
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/ |
H A D | spmd_expanders.cc | 237 REGISTER_SPMD(AssignSubVariable, TF::AssignSubVariableOp, ResourceSPMDExpander);
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
H A D | tpu_cluster_formation.mlir | 353 // CHECK-NEXT: "tf.AssignSubVariableOp"([[UNUSED_RESOURCE]], [[CONST]]) 362 …"tf.AssignSubVariableOp"(%arg1, %0) : (tensor<*x!tf_type.resource<tensor<f32>>>, tensor<f32>) -> ()
|
H A D | decompose_resource_ops.mlir | 115 // 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 D | gpu_supported_ops.md | 22 `AssignSubVariableOp` | `dtype={complex64,double,float,int32,int64,uint32,uint64}`
|
H A D | cpu_supported_ops.md | 22 `AssignSubVariableOp` | `dtype={complex64,double,float,int32,int64,uint32,uint64}`
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/g3doc/guide/ |
H A D | op_select_allowlist.md | 47 * `raw_ops.AssignSubVariableOp`
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
H A D | legalization.md | 1445 ### tf.AssignSubVariableOp 1452 %output = tf.AssignSubVariableOp(%resource, %value, %dtype)
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
H A D | tensorflow.raw_ops.pbtxt | 268 name: "AssignSubVariableOp"
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
H A D | tensorflow.raw_ops.pbtxt | 268 name: "AssignSubVariableOp"
|
/aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
H A D | wrappers.go | 1598 func AssignSubVariableOp(scope *Scope, resource tf.Output, value tf.Output) (o *tf.Operation) { func
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
H A D | ops.pbtxt | 2918 name: "AssignSubVariableOp"
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
H A D | tf_generated_ops.td | 796 def TF_AssignSubVariableOp : TF_Op<"AssignSubVariableOp", []> {
|
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ |
H A D | ops_history.v2.pbtxt | 7343 name: "AssignSubVariableOp"
|