| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/ |
| reshape_test.py | 57 shape_in, shape_out, feed_dict = self.build_shapes([3, 2], [6,])
59 event_shape_out=shape_out,
82 shape_in, shape_out, feed_dict = self.build_shapes(shape_in_static,
85 event_shape_out=shape_out,
94 bijector.inverse_event_shape_tensor(shape_out),
108 shape_in, shape_out, feed_dict = self.build_shapes([], [1,])
112 event_shape_in=shape_out, validate_args=True)
131 shape_in, shape_out, feed_dict = self.build_shapes([2, 3], [4, -1, -1,])
133 event_shape_out=shape_out,
147 shape_in, shape_out, feed_dict = self.build_shapes([2, 3], [1, 2, -2,]
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/ops/ |
| sparse_feature_cross_op.py | 97 indices_out, values_out, shape_out = (
110 indices_out, values_out, shape_out = (
122 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
|
| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_cross_op.cc | 322 Tensor* shape_out; variable
326 &shape_out, &output_start_indices);
507 Tensor** values_out, Tensor** shape_out,
528 context->allocate_output(2, TensorShape({2}), shape_out));
531 auto shape_vec = (*shape_out)->vec<int64>();
|
| /external/tensorflow/tensorflow/python/client/ |
| session_test.py | 546 indices_out, values_out, shape_out = sp_out
549 self.assertAllEqual(shape_out, shape)
556 indices_out, values_out, shape_out = s.run(sp)
559 self.assertAllEqual(shape_out, shape)
561 (indices_out, values_out, shape_out), = s.run([sp])
564 self.assertAllEqual(shape_out, shape)
571 indices_out, values_out, shape_out = s.run({'sp': sp})['sp']
574 self.assertAllEqual(shape_out, shape)
576 (indices_out, values_out, shape_out), = s.run({'sp': [sp]})['sp']
579 self.assertAllEqual(shape_out, shape
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/kernels/ |
| sparse_feature_cross_kernel.cc | 365 Tensor* shape_out; variable
369 &shape_out, &output_start_indices);
550 Tensor** values_out, Tensor** shape_out,
571 context->allocate_output(2, TensorShape({2}), shape_out));
574 auto shape_vec = (*shape_out)->vec<int64>();
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_properties_test.cc | 397 auto shape_out = properties.GetOutputProperties("while/Exit_1").at(0).shape(); local
399 EXPECT_GE(-2, shape_out.dim(0).size());
400 EXPECT_NE(shape_in.dim(0).size(), shape_out.dim(0).size());
[all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| sparse_ops.py | 435 indices_out, values_out, shape_out = gen_sparse_ops._sparse_cross(
447 return sparse_tensor.SparseTensor(indices_out, values_out, shape_out)
[all...] |
