Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/python/kernel_tests/distributions/
Dnormal_test.py120 self.evaluate(normal.batch_shape_tensor()), log_pdf.get_shape())
122 self.evaluate(normal.batch_shape_tensor()),
129 self.evaluate(normal.batch_shape_tensor()), pdf.get_shape())
131 self.evaluate(normal.batch_shape_tensor()),
156 self.evaluate(normal.batch_shape_tensor()), log_pdf.get_shape())
158 self.evaluate(normal.batch_shape_tensor()),
167 self.evaluate(normal.batch_shape_tensor()), pdf.get_shape())
169 self.evaluate(normal.batch_shape_tensor()), pdf_values.shape)
190 self.evaluate(normal.batch_shape_tensor()), cdf.get_shape())
192 self.evaluate(normal.batch_shape_tensor()),
[all …]
/external/tensorflow/tensorflow/python/ops/distributions/
Dstudent_t.py256 shape = array_ops.concat([[n], self.batch_shape_tensor()], 0)
258 df = self.df * array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)
290 v = array_ops.ones(self.batch_shape_tensor(),
306 mean = self.loc * array_ops.ones(self.batch_shape_tensor(),
313 array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)),
314 mean, array_ops.fill(self.batch_shape_tensor(), nan, name="nan"))
341 var = (array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype) *
346 self.df > array_ops.fill(self.batch_shape_tensor(), 2.), var,
347 array_ops.fill(self.batch_shape_tensor(), inf, name="inf"))
354 array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)),
[all …]
Duniform.py171 shape = array_ops.concat([[n], self.batch_shape_tensor()], 0)
179 self.batch_shape_tensor(), dtype=x.dtype)
190 array_ops.shape(x), self.batch_shape_tensor())
Dtransformed_distribution.py366 self.distribution.batch_shape_tensor())
553 self.distribution.batch_shape_tensor()
558 _ones_like(self.distribution.batch_shape_tensor())
563 [self.batch_shape_tensor(), self.event_shape_tensor()],
/external/tensorflow/tensorflow/python/ops/linalg/
Dlinear_operator_identity.py95 [self.batch_shape_tensor(),
360 special_shape = array_ops.concat((self.batch_shape_tensor(), [1, 1]), 0)
373 return array_ops.ones(shape=self.batch_shape_tensor(), dtype=self.dtype)
376 return array_ops.zeros(shape=self.batch_shape_tensor(), dtype=self.dtype)
387 shape=self.batch_shape_tensor(), dtype=self.dtype)
418 return array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)
720 shape=self.batch_shape_tensor(), dtype=self.dtype)
Dlinear_operator_householder.py229 return -array_ops.ones(shape=self.batch_shape_tensor(), dtype=self.dtype)
233 return array_ops.zeros(shape=self.batch_shape_tensor(), dtype=self.dtype)
270 return array_ops.ones(self.batch_shape_tensor(), dtype=self.dtype)
Dlinear_operator_zeros.py306 special_shape = array_ops.concat((self.batch_shape_tensor(), [1, 1]), 0)
340 return array_ops.zeros(shape=self.batch_shape_tensor(), dtype=self.dtype)
347 return array_ops.zeros(shape=self.batch_shape_tensor(), dtype=self.dtype)
467 [self.batch_shape_tensor(),
Dlinear_operator_kronecker.py281 batch_shape = self.operators[0].batch_shape_tensor()
284 batch_shape, operator.batch_shape_tensor())
331 [array_ops.ones_like(self.batch_shape_tensor()), [1, 1]], 0)
447 [array_ops.ones_like(self.batch_shape_tensor()), [1, 1]], 0)
Dlinear_operator_composition.py236 zeros = array_ops.zeros(shape=self.operators[0].batch_shape_tensor())
238 zeros += array_ops.zeros(shape=operator.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)
/external/tensorflow/tensorflow/python/ops/linalg/sparse/
Dconjugate_gradient.py113 operator.batch_shape_tensor())
117 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.-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.-exponential.pbtxt53 name: "batch_shape_tensor"
54 …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.-categorical.pbtxt56 name: "batch_shape_tensor"
57 …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.-student-t.pbtxt56 name: "batch_shape_tensor"
57 …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.-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.-normal.pbtxt52 name: "batch_shape_tensor"
53 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.linalg.-linear-operator.pbtxt109 name: "batch_shape_tensor"
110 …spec: "args=[\'self\', \'name\'], varargs=None, keywords=None, defaults=[\'batch_shape_tensor\'], "

1234