HomeSort by relevance Sort by last modified time
    Searched full:shape_in (Results 1 - 2 of 2) sorted by null

  /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,])
60 event_shape_in=shape_in,
82 shape_in, shape_out, feed_dict = self.build_shapes(shape_in_static,
86 event_shape_in=shape_in, validate_args=True)
93 bijector.forward_event_shape_tensor(shape_in),
108 shape_in, shape_out, feed_dict = self.build_shapes([], [1,])
111 event_shape_out=shape_in,
131 shape_in, shape_out, feed_dict = self.build_shapes([2, 3], [4, -1, -1,])
134 event_shape_in=shape_in,
139 sess.run(bijector.forward_event_shape_tensor(shape_in),
    [all...]
  /external/tensorflow/tensorflow/core/grappler/costs/
graph_properties_test.cc 396 auto shape_in = properties.GetOutputProperties("ones").at(0).shape(); local
398 EXPECT_GE(-2, shape_in.dim(0).size());
400 EXPECT_NE(shape_in.dim(0).size(), shape_out.dim(0).size());
    [all...]

Completed in 511 milliseconds