| /external/tensorflow/tensorflow/python/distribute/ |
| D | checkpointing_test.py | 40 variable_shape = [5]
42 array_ops.ones(variable_shape)))
55 self.assertAllClose(array_ops.ones(variable_shape), v)
56 v.assign(array_ops.zeros(variable_shape))
63 self.assertAllClose(array_ops.zeros(variable_shape), v)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | resource_variable_util.cc | 22 Status ValidateAssignUpdateVariableOpShapes(const TensorShape& variable_shape, in ValidateAssignUpdateVariableOpShapes() argument
24 if (!variable_shape.IsSameSize(value_shape)) { in ValidateAssignUpdateVariableOpShapes()
26 "Cannot update variable with shape ", variable_shape.DebugString(), in ValidateAssignUpdateVariableOpShapes()
|
| D | resource_variable_util.h | 23 Status ValidateAssignUpdateVariableOpShapes(const TensorShape& variable_shape,
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu_embedding_base.py | 84 variable_shape = (table.vocabulary_size, table.dim)
91 initializer, variable_shape, dtype=dtype)
95 shape=variable_shape,
106 shape=variable_shape,
|
| D | feature_column_v2_test.py | 94 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
110 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
330 self.assertEqual((embedding_dimension,), embedding_column_a.variable_shape)
331 self.assertEqual((embedding_dimension,), embedding_column_b.variable_shape)
363 self.assertEqual((embedding_dimension,), embedding_column_a.variable_shape)
364 self.assertEqual((embedding_dimension,), embedding_column_b.variable_shape)
|
| D | tpu_embedding_v2.py | 809 variable_shape = (table.vocabulary_size, table.dim)
815 initial_value = functools.partial(initializer, variable_shape,
820 shape=variable_shape,
831 shape=variable_shape,
|
| /external/federated-compute/fcp/artifact_building/ |
| D | variable_helpers.py | 56 variable_shape = []
60 variable_shape.append(None)
63 variable_shape.append(shape)
68 shape=variable_shape,
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | sequence_feature_column.py | 424 def variable_shape(self): member in SequenceNumericColumn
442 [array_ops.shape(dense_tensor)[:1], [-1], self.variable_shape],
451 num_elements = self.variable_shape.num_elements()
|
| D | feature_column_v2.py | 2188 def variable_shape(self): member in DenseColumn
2249 num_elements = column.variable_shape.num_elements()
2669 def variable_shape(self): member in NumericColumn
2677 return self.variable_shape
2776 def variable_shape(self): member in BucketizedColumn
2785 return self.variable_shape
2949 def variable_shape(self): member in EmbeddingColumn
2957 return self.variable_shape
3266 def variable_shape(self): member in SharedEmbeddingColumn
4239 self.variable_shape[-1])
[all …]
|
| D | feature_column_v2_test.py | 522 self.assertAllEqual((2, 3), b.variable_shape)
655 self.assertAllEqual(a_bucketized_copy.variable_shape, (2, 3))
1541 def variable_shape(self): member in OldLinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
1542 return self.variable_shape
1546 return self.variable_shape
4354 self.assertEqual(indicator_a.variable_shape, [1, 4])
4361 self.assertEqual(indicator_b.variable_shape, [1, 100])
4426 self.assertEqual(column.variable_shape, [1, 4])
4685 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
4720 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 149 TensorShapeProto variable_shape; in RewriteIntoTempVariable() local
150 variable_shape.add_dim()->set_size(0); in RewriteIntoTempVariable()
152 .Attr("shape", variable_shape) in RewriteIntoTempVariable()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | variable_ops.cc | 35 TensorShape variable_shape; in ValidateAssignUpdateVariableOpShapes() local
38 ctx->GetVariableTypeAndShape(0, &variable_dtype, &variable_shape)); in ValidateAssignUpdateVariableOpShapes()
40 ValidateAssignUpdateVariableOpShapes(variable_shape, value_shape)); in ValidateAssignUpdateVariableOpShapes()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | while_v2_indexed_slices_rewriter.py | 190 forward_shape = gen_resource_variable_ops.variable_shape(forward_input)
204 shape = gen_resource_variable_ops.variable_shape(
|
| D | control_flow_state.py | 675 resource_variable_ops.variable_shape(val),
795 gen_resource_variable_ops.variable_shape(switch_val),
811 gen_resource_variable_ops.variable_shape(val),
|
| D | resource_variable_ops.py | 2259 def variable_shape(handle, out_type=dtypes.int32): function
2262 return gen_resource_variable_ops.variable_shape(handle, out_type=out_type)
2265 return gen_resource_variable_ops.variable_shape(handle, out_type=out_type)
2275 params_shape = variable_shape(handle)
|
| D | gradients_util.py | 814 resource_variable_ops.variable_shape(t), dtype=t_dtype)
|
| D | while_v2.py | 609 gen_resource_variable_ops.variable_shape(op_output),
|
| D | array_grad.py | 746 ref_shape = gen_resource_variable_ops.variable_shape(ref, indices.dtype)
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.__internal__.feature_column.-dense-column.pbtxt | 19 name: "variable_shape"
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 126 variable_shape = tensor_shape.TensorShape(shape)
139 shape=variable_shape if variable_shape else None)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | metrics.py | 2181 variable_shape = tensor_shape.TensorShape(
2185 variable_shape = tensor_shape.TensorShape(
2191 shape=variable_shape,
2195 shape=variable_shape,
2199 shape=variable_shape,
2203 shape=variable_shape,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | sharding_util_ops_test.py | 336 variable_shape = [] if output_shape is None else output_shape
338 initial_value=np.zeros(variable_shape, dtype=input_dtype),
|
| D | variable_ops_test.py | 129 resource_variable_ops.variable_shape(h),
130 resource_variable_ops.variable_shape(h, out_type=dtypes.int64)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer_test.py | 377 var_shape = resource_variable_ops.variable_shape(var.handle)
391 new_var_shape = resource_variable_ops.variable_shape(imported_var)
|
| /external/tensorflow/tensorflow/python/kernel_tests/variables/ |
| D | resource_variable_ops_test.py | 201 vshape = resource_variable_ops.variable_shape(v.handle)
|