Home
last modified time | relevance | path

Searched refs:batch_shape_tensor (Results 1 – 25 of 94) sorted by relevance

1234

/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dnormal_test.py116 self.evaluate(normal.batch_shape_tensor()), log_pdf.get_shape())
118 self.evaluate(normal.batch_shape_tensor()),
125 self.evaluate(normal.batch_shape_tensor()), pdf.get_shape())
127 self.evaluate(normal.batch_shape_tensor()),
152 self.evaluate(normal.batch_shape_tensor()), log_pdf.get_shape())
154 self.evaluate(normal.batch_shape_tensor()),
163 self.evaluate(normal.batch_shape_tensor()), pdf.get_shape())
165 self.evaluate(normal.batch_shape_tensor()), pdf_values.shape)
186 self.evaluate(normal.batch_shape_tensor()), cdf.get_shape())
188 self.evaluate(normal.batch_shape_tensor()),
[all …]
/external/tensorflow/tensorflow/python/ops/distributions/
Dstudent_t.py252 shape = array_ops.concat([[n], self.batch_shape_tensor()], 0)
254 df = self.df * array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)
286 v = array_ops.ones(self.batch_shape_tensor(),
302 mean = self.loc * array_ops.ones(self.batch_shape_tensor(),
309 array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)),
310 mean, array_ops.fill(self.batch_shape_tensor(), nan, name="nan"))
337 var = (array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype) *
342 self.df > array_ops.fill(self.batch_shape_tensor(), 2.), var,
343 array_ops.fill(self.batch_shape_tensor(), inf, name="inf"))
350 array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)),
[all …]
Duniform.py167 shape = array_ops.concat([[n], self.batch_shape_tensor()], 0)
175 self.batch_shape_tensor(), dtype=x.dtype)
186 array_ops.shape(x), self.batch_shape_tensor())
Dtransformed_distribution.py362 self.distribution.batch_shape_tensor())
549 self.distribution.batch_shape_tensor()
554 _ones_like(self.distribution.batch_shape_tensor())
559 [self.batch_shape_tensor(), self.event_shape_tensor()],
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_identity.py91 [self.batch_shape_tensor(),
357 special_shape = array_ops.concat((self.batch_shape_tensor(), [1, 1]), 0)
370 return array_ops.ones(shape=self.batch_shape_tensor(), dtype=self.dtype)
373 return array_ops.zeros(shape=self.batch_shape_tensor(), dtype=self.dtype)
384 shape=self.batch_shape_tensor(), dtype=self.dtype)
415 return array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)
734 shape=self.batch_shape_tensor(), dtype=self.dtype)
Dlinear_operator_householder.py224 …return -array_ops.ones(shape=self.batch_shape_tensor(), dtype=self.dtype) # pylint: disable=inval…
228 return array_ops.zeros(shape=self.batch_shape_tensor(), dtype=self.dtype)
263 return array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)
Dlinear_operator_zeros.py303 special_shape = array_ops.concat((self.batch_shape_tensor(), [1, 1]), 0)
337 return array_ops.zeros(shape=self.batch_shape_tensor(), dtype=self.dtype)
344 return array_ops.zeros(shape=self.batch_shape_tensor(), dtype=self.dtype)
464 [self.batch_shape_tensor(),
Dslicing.py128 param_batch_shape = param.batch_shape_tensor()
172 batch_shape = linop.batch_shape_tensor()
Dlinear_operator_permutation.py237 return array_ops.zeros(shape=self.batch_shape_tensor(), dtype=self.dtype)
258 return array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)
Dlinear_operator_composition.py229 zeros = array_ops.zeros(shape=self.operators[0].batch_shape_tensor())
231 zeros += array_ops.zeros(shape=operator.batch_shape_tensor())
/external/tensorflow/tensorflow/python/kernel_tests/linalg/
Dslicing_test.py197 sliced.batch_shape_tensor())
207 self.assertAllEqual(linop[0, ...].batch_shape_tensor(), [3, 5, 2])
209 0, ..., array_ops.newaxis].batch_shape_tensor(), [3, 5, 2, 1])
211 ..., array_ops.newaxis].batch_shape_tensor(), [4, 3, 5, 2, 1])
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dconjugate_gradient.py109 operator.batch_shape_tensor())
113 preconditioner.batch_shape_tensor()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distributions.-distribution.pbtxt43 name: "batch_shape_tensor"
44 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-laplace.pbtxt52 name: "batch_shape_tensor"
53 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-multinomial.pbtxt56 name: "batch_shape_tensor"
57 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-dirichlet.pbtxt52 name: "batch_shape_tensor"
53 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-gamma.pbtxt52 name: "batch_shape_tensor"
53 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-bernoulli.pbtxt52 name: "batch_shape_tensor"
53 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-dirichlet-multinomial.pbtxt56 name: "batch_shape_tensor"
57 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-beta.pbtxt56 name: "batch_shape_tensor"
57 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-uniform.pbtxt52 name: "batch_shape_tensor"
53 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-normal.pbtxt52 name: "batch_shape_tensor"
53 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-student-t.pbtxt56 name: "batch_shape_tensor"
57 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-exponential.pbtxt53 name: "batch_shape_tensor"
54 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
Dtensorflow.distributions.-categorical.pbtxt56 name: "batch_shape_tensor"
57 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "

1234