Home
last modified time | relevance | path

Searched refs:validate_shape (Results 1 – 25 of 63) sorted by relevance

123

/external/tensorflow/tensorflow/python/training/
Dslot_creator.py50 validate_shape, argument
77 validate_shape=validate_shape)
144 validate_shape = val.get_shape().is_fully_defined()
157 validate_shape,
166 validate_shape,
203 validate_shape = shape.is_fully_defined()
216 validate_shape,
225 validate_shape,
/external/tensorflow/tensorflow/python/ops/
Dvariable_scope.py327 validate_shape=True, argument
476 validate_shape=True, argument
501 validate_shape=validate_shape,
522 validate_shape=validate_shape,
545 validate_shape=validate_shape,
570 "validate_shape": validate_shape,
593 validate_shape=validate_shape,
610 validate_shape=True, argument
803 validate_shape=validate_shape,
837 validate_shape=True, argument
[all …]
Daccumulate_n_benchmark.py36 def _AccumulateNTemplate(self, inputs, init, shape, validate_shape): argument
39 ref = state_ops.assign(var, init, validate_shape=validate_shape)
52 validate_shape=True)
59 validate_shape=False)
67 validate_shape=True)
Dvariables.py190 validate_shape=True, argument
214 validate_shape=validate_shape,
230 validate_shape=True, argument
251 validate_shape=validate_shape,
375 validate_shape=True, argument
1458 validate_shape=True, argument
1556 validate_shape=True, argument
1652 validate_shape=validate_shape,
1675 validate_shape=True, argument
1801 if validate_shape else tensor_shape.unknown_shape())
[all …]
Dclustering_ops.py292 init_value, name=CLUSTERS_VAR_NAME, validate_shape=False)
300 init_value, name='clusters_updated', validate_shape=False)
694 self._cluster_centers, new_centers, validate_shape=False)
699 validate_shape=False)
749 self._cluster_centers, all_centers, validate_shape=False)
752 self._cluster_centers_updated, a, validate_shape=False)
Dresource_variable_ops.py359 validate_shape=True, argument
445 self._validate_shape = validate_shape
941 validate_shape = self._validate_shape and self._shape.is_fully_defined()
942 kwargs["validate_shape"] = validate_shape
1572 validate_shape=True, # pylint: disable=unused-argument argument
1661 validate_shape=validate_shape)
1675 validate_shape=validate_shape,
1691 validate_shape=True, argument
1933 validate_shape=validate_shape,
1939 validate_shape=True): argument
[all …]
Dstate_ops.py277 def assign(ref, value, validate_shape=None, use_locking=None, name=None): argument
354 validate_shape=validate_shape)
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dvariable_scope_shim.py165 validate_shape=True, argument
308 validate_shape=True, argument
335 validate_shape=validate_shape,
359 "validate_shape": validate_shape,
382 validate_shape=validate_shape,
399 validate_shape=True, argument
504 validate_shape=validate_shape,
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor.h73 validate_shape(true) {} in ImportGraphDefOptions()
147 bool validate_shape; member
/external/tensorflow/tensorflow/python/kernel_tests/v1_compat_tests/
Ddense_update_ops_test.py36 a = state_ops.assign(p, data, validate_shape=False)
42 a2 = state_ops.assign(p, data2, validate_shape=False)
/external/tensorflow/tensorflow/python/kernel_tests/variables/
Dvariable_ops_test.py90 assigned = state_ops.assign(var, value, validate_shape=False)
106 assigned = state_ops.assign(var, value, validate_shape=False)
129 assigned = state_ops.assign(var, value, validate_shape=False)
149 state_ops.assign(var, value, validate_shape=False).get_shape())
284 validate_shape=False,
286 result = state_ops.assign(buffer_var, data, validate_shape=False)
/external/tensorflow/tensorflow/core/framework/
Dref_var.cc23 int value_index, bool use_locking, bool validate_shape, in AssignRefVariable() argument
52 if (validate_shape) { in AssignRefVariable()
Dref_var.h26 int value_index, bool use_locking, bool validate_shape,
/external/tensorflow/tensorflow/c/
Dkernels_experimental.h65 bool validate_shape,
78 int value_index, bool use_locking, bool validate_shape,
/external/tensorflow/tensorflow/c/experimental/ops/
Dresource_variable_ops.cc86 AbstractTensorHandle* const value, bool validate_shape, in AssignVariableOp() argument
93 TF_RETURN_IF_ERROR(op_ptr->SetAttrBool("validate_shape", validate_shape)); in AssignVariableOp()
Dresource_variable_ops.h46 bool validate_shape = false, const char* name = nullptr,
/external/tensorflow/tensorflow/core/ops/
Dstate_ops.cc88 bool validate_shape; in __anonebb9250a0202() local
89 TF_RETURN_IF_ERROR(c->GetAttr("validate_shape", &validate_shape)); in __anonebb9250a0202()
90 if (validate_shape) { in __anonebb9250a0202()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/
Dconvert_ref_variables.mlir15 …%2 = "tf.Assign"(%0, %1) {T = i32, device = "", use_locking = true, validate_shape = true} : (tens…
48 …%2 = "tf.Assign"(%1, %0) {T = i32, device = "", use_locking = true, validate_shape = true} : (tens…
56 …%5 = "tf.Assign"(%1, %4) {T = i32, device = "", use_locking = true, validate_shape = true} : (tens…
/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
Dtest3.pbtxt116 key: "validate_shape"
343 key: "validate_shape"
641 key: "validate_shape"
804 key: "validate_shape"
1151 key: "validate_shape"
1314 key: "validate_shape"
1474 key: "validate_shape"
1561 key: "validate_shape"
1648 key: "validate_shape"
1735 key: "validate_shape"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DAssign.pbtxt22 name: "validate_shape"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DAssign.pbtxt22 name: "validate_shape"
DAssignVariableOp.pbtxt32 name: "validate_shape"
/external/tensorflow/tensorflow/cc/saved_model/testdata/half_plus_two_pbtxt/00000123/
Dsaved_model.pbtxt60 name: "validate_shape"
558 key: "validate_shape"
695 key: "validate_shape"
832 key: "validate_shape"
1547 key: "validate_shape"
1646 key: "validate_shape"
2154 key: "validate_shape"
2292 key: "validate_shape"
2430 key: "validate_shape"
/external/tensorflow/tensorflow/core/ir/importexport/tests/mlir_to_graphdef/
Dfulltype.mlir12 …k:0/device:CPU:0") name("Variable/Assign") {dtype = !tf_type.string, validate_shape = false} : ten…
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_variables.td38 (TF_AssignVariableOp $resource, $value, $validate_shape),

123