Home
last modified time | relevance | path

Searched defs:validate_shape (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dvariable_scope.py327 validate_shape=True, argument
476 validate_shape=True, argument
610 validate_shape=True, argument
837 validate_shape=True, argument
1284 validate_shape=True, argument
1355 validate_shape=True, argument
1610 validate_shape=True, argument
1787 validate_shape=True, argument
1831 validate_shape=True, argument
H A Dvariables.py190 validate_shape=True, argument
230 validate_shape=True, argument
375 validate_shape=True, argument
1458 validate_shape=True, argument
1556 validate_shape=True, argument
1675 validate_shape=True, argument
H A Dresource_variable_ops.py359 validate_shape=True, argument
1572 validate_shape=True, # pylint: disable=unused-argument argument
1691 validate_shape=True, argument
1939 validate_shape=True): argument
H A Daccumulate_n_benchmark.py36 def _AccumulateNTemplate(self, inputs, init, shape, validate_shape): argument
H A Dstate_ops.py277 def assign(ref, value, validate_shape=None, use_locking=None, name=None): argument
H A Dmetrics_impl.py37 def metric_variable(shape, dtype, validate_shape=True, name=None): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
H A Dvariable_scope_shim.py165 validate_shape=True, argument
308 validate_shape=True, argument
399 validate_shape=True, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dref_var.cc23 int value_index, bool use_locking, bool validate_shape, in AssignRefVariable()
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dslot_creator.py50 validate_shape, argument
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dstate_ops.cc88 bool validate_shape; in __anon9ff5ca8a0202() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dkernels_experimental.cc172 int value_index, bool validate_shape, in TF_AssignVariable()
221 bool use_locking, bool validate_shape, in TF_AssignRefVariable()
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/ops/
H A Dresource_variable_ops.cc86 AbstractTensorHandle* const value, bool validate_shape, in AssignVariableOp()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_constructor.h147 bool validate_shape; member
H A Dgraph_constructor.cc147 bool validate_shape = true; member
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/
H A Dtpu_reshard_variables_op_util.cc113 auto validate_shape = [&](int i, const Tensor& tensor) { in BuildInputBuffers() local
H A Dtpu_execute_op.cc210 auto validate_shape = [&](int i, const Tensor& tensor) { in BuildComputationInputs() local
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dbase_layer_utils.py57 validate_shape=True, argument
/aosp_15_r20/external/pytorch/torch/_prims_common/
H A D__init__.py589 def validate_shape(shape: ShapeType): function