Home
last modified time | relevance | path

Searched refs:shape_in (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/
Dreshape_test.py54 shape_in, shape_out, feed_dict = self.build_shapes([3, 2], [6,])
57 event_shape_in=shape_in,
79 shape_in, shape_out, feed_dict = self.build_shapes(shape_in_static,
83 event_shape_in=shape_in, validate_args=True)
90 bijector.forward_event_shape_tensor(shape_in),
105 shape_in, shape_out, feed_dict = self.build_shapes([], [1,])
108 event_shape_out=shape_in,
128 shape_in, shape_out, feed_dict = self.build_shapes([2, 3], [4, -1, -1,])
131 event_shape_in=shape_in,
136 sess.run(bijector.forward_event_shape_tensor(shape_in),
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dxent_op.cc47 TensorShape shape_in = logits_in.shape(); in Compute() local
57 shape_in = BCast::ToShape(bcast.output_shape()); in Compute()
59 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(shape_in), in Compute()
69 TensorShape({shape_in.dim_size(0), 1}), in Compute()
75 0, TensorShape({shape_in.dim_size(0)}), &loss_out)); in Compute()
79 {0}, 1, shape_in, &back_out)); in Compute()
80 if (shape_in.dim_size(0) > 0) { in Compute()
83 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(), in Compute()
89 functor(context->eigen_device<Device>(), shape_in.AsEigenDSizes<2>(), in Compute()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties_test.cc581 auto shape_in = properties.GetOutputProperties("ones").at(0).shape(); in TEST_F() local
583 EXPECT_GE(-2, shape_in.dim(0).size()); in TEST_F()
585 EXPECT_NE(shape_in.dim(0).size(), shape_out.dim(0).size()); in TEST_F()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc469 ShapeHandle shape_in = c->input(1); in __anondb9326b20702() local
470 TF_RETURN_IF_ERROR(c->WithRank(shape_in, 1, &shape_in)); in __anondb9326b20702()