Home
last modified time | relevance | path

Searched refs:variable_shape (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcheckpointing_test.py45 variable_shape = [5]
47 array_ops.ones(variable_shape)))
60 self.assertAllClose(array_ops.ones(variable_shape), v)
61 v.assign(array_ops.zeros(variable_shape))
68 self.assertAllClose(array_ops.zeros(variable_shape), v)
/external/tensorflow/tensorflow/python/feature_column/
Dsequence_feature_column.py428 def variable_shape(self): member in SequenceNumericColumn
446 [array_ops.shape(dense_tensor)[:1], [-1], self.variable_shape],
455 num_elements = self.variable_shape.num_elements()
Dfeature_column_v2.py2124 def variable_shape(self): member in DenseColumn
2185 num_elements = column.variable_shape.num_elements()
2602 def variable_shape(self): member in NumericColumn
2610 return self.variable_shape
2709 def variable_shape(self): member in BucketizedColumn
2718 return self.variable_shape
2882 def variable_shape(self): member in EmbeddingColumn
2890 return self.variable_shape
3198 def variable_shape(self): member in SharedEmbeddingColumn
4131 self.variable_shape[-1])
[all …]
Dfeature_column_v2_test.py526 self.assertAllEqual((2, 3), b.variable_shape)
659 self.assertAllEqual(a_bucketized_copy.variable_shape, (2, 3))
1545 def variable_shape(self): member in OldLinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
1546 return self.variable_shape
1550 return self.variable_shape
4325 self.assertEqual(indicator_a.variable_shape, [1, 4])
4332 self.assertEqual(indicator_b.variable_shape, [1, 100])
4397 self.assertEqual(column.variable_shape, [1, 4])
4656 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
4691 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
[all …]
/external/tensorflow/tensorflow/python/keras/feature_column/
Dbase_feature_layer.py98 total_elements += column.variable_shape.num_elements()
111 num_elements = column.variable_shape.num_elements()
/external/tensorflow/tensorflow/core/common_runtime/
Daccumulate_n_optimizer.cc149 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/tpu/
Dfeature_column_v2_test.py96 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
112 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
312 self.assertEqual((embedding_dimension,), embedding_column_a.variable_shape)
313 self.assertEqual((embedding_dimension,), embedding_column_b.variable_shape)
345 self.assertEqual((embedding_dimension,), embedding_column_a.variable_shape)
346 self.assertEqual((embedding_dimension,), embedding_column_b.variable_shape)
Dtpu_embedding_v2.py800 variable_shape = (table.vocabulary_size, table.dim)
806 initial_value = functools.partial(initializer, variable_shape,
811 shape=variable_shape,
822 shape=variable_shape,
/external/tensorflow/tensorflow/python/ops/
Dwhile_v2_indexed_slices_rewriter.py191 forward_shape = gen_resource_variable_ops.variable_shape(forward_input)
205 shape = gen_resource_variable_ops.variable_shape(forward_input,
Dcontrol_flow_state.py679 resource_variable_ops.variable_shape(val),
799 gen_resource_variable_ops.variable_shape(switch_val),
815 gen_resource_variable_ops.variable_shape(val),
Dresource_variable_ops.py2144 def variable_shape(handle, out_type=dtypes.int32): function
2147 return gen_resource_variable_ops.variable_shape(handle, out_type=out_type)
2150 return gen_resource_variable_ops.variable_shape(handle, out_type=out_type)
2160 params_shape = variable_shape(handle)
Dgradients_util.py803 resource_variable_ops.variable_shape(t), dtype=t_dtype)
Dwhile_v2.py562 gen_resource_variable_ops.variable_shape(op_output),
Darray_grad.py726 ref_shape = gen_resource_variable_ops.variable_shape(ref, indices.dtype)
/external/tensorflow/tensorflow/python/keras/engine/
Dbase_layer_utils.py129 variable_shape = tensor_shape.TensorShape(shape)
142 shape=variable_shape if variable_shape else None)
/external/tensorflow/tensorflow/python/keras/
Dmetrics.py2057 variable_shape = tensor_shape.TensorShape(
2061 variable_shape = tensor_shape.TensorShape(
2067 shape=variable_shape,
2071 shape=variable_shape,
2075 shape=variable_shape,
2079 shape=variable_shape,
/external/tensorflow/tensorflow/compiler/tests/
Dvariable_ops_test.py130 resource_variable_ops.variable_shape(h),
131 resource_variable_ops.variable_shape(h, out_type=dtypes.int64)
/external/tensorflow/tensorflow/python/framework/
Dimporter_test.py381 var_shape = resource_variable_ops.variable_shape(var.handle)
395 new_var_shape = resource_variable_ops.variable_shape(imported_var)
/external/tensorflow/tensorflow/python/kernel_tests/
Dresource_variable_ops_test.py177 vshape = resource_variable_ops.variable_shape(v.handle)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py2435 return resource_variable_ops.variable_shape(v.handle)