| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column_v2_test.py | 59 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
75 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
225 self.assertEqual((embedding_dimension,), embedding_column_a.variable_shape)
226 self.assertEqual((embedding_dimension,), embedding_column_b.variable_shape)
258 self.assertEqual((embedding_dimension,), embedding_column_a.variable_shape)
259 self.assertEqual((embedding_dimension,), embedding_column_b.variable_shape)
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | sequence_feature_column.py | 543 def variable_shape(self): member in SequenceNumericColumn
561 [array_ops.shape(dense_tensor)[:1], [-1], self.variable_shape],
570 num_elements = self.variable_shape.num_elements()
|
| D | feature_column_v2.py | 431 total_elements += column.variable_shape.num_elements()
443 num_elements = column.variable_shape.num_elements()
520 first_dim = column.variable_shape.num_elements()
2427 def variable_shape(self): member in DenseColumn
2488 num_elements = column.variable_shape.num_elements()
2891 def variable_shape(self): member in NumericColumn
2899 return self.variable_shape
2996 def variable_shape(self): member in BucketizedColumn
3005 return self.variable_shape
3169 def variable_shape(self): member in EmbeddingColumn
[all …]
|
| D | feature_column_v2_test.py | 570 self.assertAllEqual((2, 3), b.variable_shape)
706 self.assertAllEqual(a_bucketized_copy.variable_shape, (2, 3))
1773 def variable_shape(self): member in LinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
2447 def variable_shape(self): member in OldLinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
2448 return self.variable_shape
2452 return self.variable_shape
5342 self.assertEqual(indicator_a.variable_shape, [1, 4])
5349 self.assertEqual(indicator_b.variable_shape, [1, 100])
5415 self.assertEqual(column.variable_shape, [1, 4])
5723 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/python/ops/ |
| D | while_v2_indexed_slices_rewriter.py | 182 forward_shape = gen_resource_variable_ops.variable_shape(forward_input)
196 shape = gen_resource_variable_ops.variable_shape(forward_input,
|
| D | control_flow_state.py | 679 resource_variable_ops.variable_shape(val),
799 gen_resource_variable_ops.variable_shape(switch_val),
815 gen_resource_variable_ops.variable_shape(val),
|
| D | resource_variable_ops.py | 1959 def variable_shape(handle, out_type=dtypes.int32): function
1962 return gen_resource_variable_ops.variable_shape(handle, out_type=out_type)
1965 return gen_resource_variable_ops.variable_shape(handle, out_type=out_type)
1975 params_shape = variable_shape(handle)
|
| D | gradients_util.py | 793 resource_variable_ops.variable_shape(t), dtype=t_dtype)
|
| D | while_v2.py | 531 gen_resource_variable_ops.variable_shape(op_output),
|
| D | array_grad.py | 698 ref_shape = gen_resource_variable_ops.variable_shape(ref, indices.dtype)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | base_layer_utils.py | 129 variable_shape = tensor_shape.TensorShape(shape)
142 shape=variable_shape if variable_shape else None)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | metrics.py | 1876 variable_shape = tensor_shape.TensorShape(
1879 variable_shape = tensor_shape.TensorShape(
1885 shape=variable_shape,
1889 shape=variable_shape,
1893 shape=variable_shape,
1897 shape=variable_shape,
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | variable_ops_test.py | 130 resource_variable_ops.variable_shape(h),
131 resource_variable_ops.variable_shape(h, out_type=dtypes.int64)
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | importer_test.py | 380 var_shape = resource_variable_ops.variable_shape(var.handle)
394 new_var_shape = resource_variable_ops.variable_shape(imported_var)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | resource_variable_ops_test.py | 172 resource_variable_ops.variable_shape(v.handle)),
|