Home
last modified time | relevance | path

Searched refs:device_var_updates_indices (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtpu_colocate_composite_resource_ops.mlir22 …ndUpdateVariables"(%arg1, %1) {device_var_reads_indices = [0], device_var_updates_indices = [-1]} …
53 …ndUpdateVariables"(%arg1, %2) {device_var_reads_indices = [0], device_var_updates_indices = [-1]} …
87 …ndUpdateVariables"(%arg1, %2) {device_var_reads_indices = [0], device_var_updates_indices = [-1]} …
112 …ndUpdateVariables"(%arg1, %2) {device_var_reads_indices = [0], device_var_updates_indices = [-1]} …
Dtpu-variable-runtime-reformatting.mlir70 {device_var_reads_indices = [0, 1], device_var_updates_indices = [0, 1]}
147 {device_var_reads_indices = [0, 1, 2], device_var_updates_indices = [0, 1, 2]}
207 {device_var_reads_indices = [0, 1], device_var_updates_indices = [0, 1]}
290 {device_var_reads_indices = [0, 1], device_var_updates_indices = [0, 1]}
Dtpu-merge-variables-with-execute.mlir35 // CHECK-SAME: device_var_updates_indices = [0, -1]
81 // CHECK-SAME: device_var_updates_indices = [0]
142 // CHECK-SAME: device_var_updates_indices = [1, -1]
Dtf-ops.mlir4139 …teVariables"(%arg0, %arg1) {device_var_reads_indices = [0, 1], device_var_updates_indices = [0]} :…
4146 …// expected-error@below {{requires 'device_var_updates_indices' to be the same size as number of r…
4147 …pdateVariables"(%arg0, %arg1) {device_var_reads_indices = [0], device_var_updates_indices = [0, 1]…
4155 …dateVariables"(%arg0, %arg1) {device_var_reads_indices = [-1], device_var_updates_indices = [0]} :…
4162 …// expected-error@below {{requires 'device_var_updates_indices' to contain values of at least -1, …
4163 …pdateVariables"(%arg0, %arg1) {device_var_reads_indices = [0], device_var_updates_indices = [-2]} …
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUExecuteAndUpdateVariables.pbtxt31 name: "device_var_updates_indices"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_TPUExecuteAndUpdateVariables.pbtxt8 indices from directly reading variables) and device_var_updates_indices (program
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_merge_variables_with_execute.cc518 llvm::SmallVector<int64_t, 8> device_var_updates_indices; in MergeForOneTPUExecute() local
522 device_var_updates_indices.push_back(info.execute_output_index); in MergeForOneTPUExecute()
531 !llvm::is_contained(device_var_updates_indices, it.index())) { in MergeForOneTPUExecute()
549 builder->getI64ArrayAttr(device_var_updates_indices))}); in MergeForOneTPUExecute()
Dtpu_variable_runtime_reformatting.cc256 int64_t ret_index = execute.device_var_updates_indices() in AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc2140 failed(check_attr(op.device_var_updates_indices(), in Verify()
2164 if (device_var_updates_indices() in getEffects()
Dtf_generated_ops.td15744 indices from directly reading variables) and device_var_updates_indices (program
15755 I64ArrayAttr:$device_var_updates_indices
/external/tensorflow/tensorflow/go/op/
Dwrappers.go30090 …put, Tresults []tf.DataType, device_var_reads_indices []int64, device_var_updates_indices []int64)…
30094 …reads_indices": device_var_reads_indices, "device_var_updates_indices": device_var_updates_indices}
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt53856 name: "device_var_updates_indices"