| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | accumulate_n_optimizer.cc | 117 TensorShapeProto variable_shape; in rewriteNode() local
118 variable_shape.add_dim()->set_size(0); in rewriteNode()
120 .Attr("shape", variable_shape) in rewriteNode()
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | sequence_feature_column.py | 538 def variable_shape(self): member in SequenceNumericColumn
556 [array_ops.shape(dense_tensor)[:1], [-1], self.variable_shape],
565 num_elements = self.variable_shape.num_elements()
|
| D | feature_column_v2.py | 362 total_elements += column.variable_shape.num_elements()
374 num_elements = column.variable_shape.num_elements()
523 first_dim = column.variable_shape.num_elements()
2311 def variable_shape(self): member in DenseColumn
2372 num_elements = column.variable_shape.num_elements()
2775 def variable_shape(self): member in NumericColumn
2783 return self.variable_shape
2878 def variable_shape(self): member in BucketizedColumn
2887 return self.variable_shape
3026 def variable_shape(self): member in EmbeddingColumn
[all …]
|
| D | feature_column_v2_test.py | 524 self.assertAllEqual((2, 3), b.variable_shape)
660 self.assertAllEqual(a_bucketized_copy.variable_shape, (2, 3))
1720 def variable_shape(self): member in LinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
2434 def variable_shape(self): member in OldLinearModelTest.test_dense_and_sparse_column._DenseAndSparseColumn
2435 return self.variable_shape
2439 return self.variable_shape
5878 self.assertEqual(indicator_a.variable_shape, [1, 4])
5885 self.assertEqual(indicator_b.variable_shape, [1, 100])
5947 self.assertEqual(column.variable_shape, [1, 4])
6252 self.assertEqual((embedding_dimension,), embedding_column.variable_shape)
[all …]
|
| /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 | resource_variable_ops.py | 1498 def variable_shape(handle, out_type=dtypes.int32): function
1501 return gen_resource_variable_ops.variable_shape(handle, out_type=out_type)
1504 return gen_resource_variable_ops.variable_shape(handle, out_type=out_type)
1514 params_shape = variable_shape(handle)
|
| D | cond_v2.py | 302 gen_resource_variable_ops.variable_shape(func_graph.inputs[i]))
|
| D | while_v2.py | 373 gen_resource_variable_ops.variable_shape(op_output))
|
| D | control_flow_ops.py | 1329 return array_ops.zeros(gen_resource_variable_ops.variable_shape(val))
1344 gen_resource_variable_ops.variable_shape(switch_val))
|
| /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 | 157 resource_variable_ops.variable_shape(v.handle)),
|