Home
last modified time | relevance | path

Searched refs:unknown_shape (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc153 absl::optional<std::vector<int64_t>> unknown_shape() const { in unknown_shape() function in tensorflow::__anon347214ea0111::ShapeInferenceTest
177 /*input_shapes*/ {make_shape({3, 2}), unknown_shape()}, in TEST_F()
190 /*input_shapes*/ {unknown_shape(), unknown_shape()}, in TEST_F()
210 /* input_shapes*/ {unknown_shape(), unknown_shape()}, in TEST_F()
226 /* input_shapes*/ {unknown_shape(), unknown_shape()}, in TEST_F()
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_ops_test.py78 self.assertEqual(tensor_shape.unknown_shape(), p.get_shape())
100 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
108 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
114 self.assertEqual(tensor_shape.unknown_shape(), tensor.get_shape())
133 self.assertEqual(tensor_shape.unknown_shape(), assigned.get_shape())
140 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
141 self.assertEqual(tensor_shape.unknown_shape(),
149 self.assertEqual(tensor_shape.unknown_shape(), var.get_shape())
151 tensor_shape.unknown_shape(),
183 self.assertEqual(tensor_shape.unknown_shape(), added.get_shape())
[all …]
Dstack_ops_test.py96 c1, b1, [r, v], [r.get_shape(), tensor_shape.unknown_shape()])
248 c1, b1, [r, v], [r.get_shape(), tensor_shape.unknown_shape()])
Dcontrol_flow_ops_py_test.py1916 tensor_shape.unknown_shape(),
1917 tensor_shape.unknown_shape()
2007 [i.get_shape(), tensor_shape.unknown_shape()])
2878 [r.get_shape(), tensor_shape.unknown_shape()],
2946 [n.get_shape(), tensor_shape.unknown_shape()],
3329 [tensor_shape.unknown_shape()])
3364 [tensor_shape.unknown_shape()])
3368 [tensor_shape.unknown_shape()])
4563 [tensor_shape.unknown_shape(), tensor_shape.unknown_shape()])
4578 [tensor_shape.unknown_shape(),
[all …]
Dreduction_ops_test.py328 self.assertEqual(tensor_shape.unknown_shape(), s_unknown.get_shape())
335 c_known_rank.set_shape(tensor_shape.unknown_shape(rank=3))
348 self.assertEqual(tensor_shape.unknown_shape(),
Dtensor_array_ops_test.py997 shape_invariants=(time_0.get_shape(), tensor_shape.unknown_shape(),
998 tensor_shape.unknown_shape()),
1343 self.assertAllEqual(r0.get_shape(), tensor_shape.unknown_shape())
1740 self.assertEqual(tensor_shape.unknown_shape(), read0.get_shape())
/external/tensorflow/tensorflow/python/data/util/
Dsparse_test.py97 "expected": tensor_shape.unknown_shape()
107 "expected": (tensor_shape.unknown_shape())
122 "expected": (tensor_shape.unknown_shape(), ())
127 "expected": ((), tensor_shape.unknown_shape())
141 "expected": (tensor_shape.unknown_shape(), (),
142 tensor_shape.unknown_shape())
152 "expected": ((), tensor_shape.unknown_shape(), ())
Dsparse.py53 tensor_shape.unknown_shape() if c is sparse_tensor.SparseTensor else shape
/external/tensorflow/tensorflow/python/framework/
Dtensor_shape_test.py430 self.assertEqual("<unknown>", str(tensor_shape.unknown_shape()))
433 str(tensor_shape.unknown_shape(rank=1)).replace("?", "None"))
436 str(tensor_shape.unknown_shape(rank=2)).replace("?", "None"))
439 str(tensor_shape.unknown_shape(rank=3)).replace("?", "None"))
449 self.assertTrue(tensor_shape.unknown_shape().as_proto().unknown_rank)
451 tensor_shape.unknown_shape(rank=3).as_proto().unknown_rank)
473 unk0 = tensor_shape.unknown_shape()
481 unk1 = tensor_shape.unknown_shape()
494 tensor_shape.unknown_shape().as_list()
495 self.assertAllEqual([None, None], tensor_shape.unknown_shape(2).as_list())
Dtensor_shape.py909 return unknown_shape()
913 return unknown_shape()
915 return unknown_shape(rank=stop - start)
1008 return unknown_shape()
1056 return self.merge_with(unknown_shape(rank=rank))
1182 return unknown_shape()
1305 def unknown_shape(rank=None, **kwargs): function
Dcommon_shapes_test.py69 unknown = tensor_shape.unknown_shape()
92 unknown = tensor_shape.unknown_shape()
157 unknown = tensor_shape.unknown_shape()
Dcommon_shapes.py103 return tensor_shape.unknown_shape()
Dtensor_util.py920 return tensor_shape.unknown_shape()
932 return tensor_shape.unknown_shape(shape.dims[0].value)
1022 ret = tensor_shape.unknown_shape(shape.dims[0].value)
Dsparse_tensor.py282 return SparseTensorSpec(tensor_shape.unknown_shape(rank), self.dtype)
Dops.py460 shape_vec, unknown_shape = pywrap_tf_session.TF_GraphGetTensorShapeHelper(
462 if unknown_shape:
463 return tensor_shape.unknown_shape()
757 unknown_shape = True
759 unknown_shape = False
770 unknown_shape)
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.h141 TF_Graph* graph, TF_Output output, TF_Status* status, bool* unknown_shape);
242 bool unknown_shape, TF_Status* status);
Dtf_session_helper.cc486 bool* unknown_shape) { in TF_GraphGetTensorShapeHelper() argument
489 *unknown_shape = false; in TF_GraphGetTensorShapeHelper()
496 *unknown_shape = true; in TF_GraphGetTensorShapeHelper()
679 bool unknown_shape, TF_Status* status) { in TF_GraphSetTensorShape_wrapper() argument
680 if (unknown_shape) { in TF_GraphSetTensorShape_wrapper()
Dtf_session_wrapper.cc226 bool unknown_shape; in PYBIND11_MODULE() local
229 graph, output, status.get(), &unknown_shape); in PYBIND11_MODULE()
239 py::tuple result_tuple = py::make_tuple(py_list, py::cast(unknown_shape)); in PYBIND11_MODULE()
245 bool unknown_shape) { in PYBIND11_MODULE() argument
252 graph, output, dims, unknown_shape, status.get()); in PYBIND11_MODULE()
258 bool unknown_shape) { in PYBIND11_MODULE() argument
264 graph, output, dims, unknown_shape, status.get()); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/core/ops/
Dmath_ops_test.cc248 PartialTensorShape unknown_shape; in TEST() local
252 new ShapeDtypeV{{scalar, DT_FLOAT}, {unknown_shape, DT_INT32}}); in TEST()
255 new ShapeDtypeV{{i1, DT_FLOAT}, {unknown_shape, DT_INT32}}); in TEST()
/external/tensorflow/tensorflow/python/ops/
Ddata_flow_ops.py107 return tensor_shape.unknown_shape()
166 self._shapes = [tensor_shape.unknown_shape() for _ in self._dtypes]
1065 self._shapes = [tensor_shape.unknown_shape() for _ in self._types]
1263 self._shape = tensor_shape.unknown_shape()
1633 self._shapes = [tensor_shape.unknown_shape() for _ in self._dtypes]
Dstate_ops.py45 shape = tensor_shape.unknown_shape()
Dembedding_ops.py937 tensor_shape.unknown_shape(
Dvariables.py1807 if validate_shape else tensor_shape.unknown_shape())
1827 if validate_shape else tensor_shape.unknown_shape())
/external/tensorflow/tensorflow/python/saved_model/
Dutils_impl.py138 tensor_shape=tensor_shape.unknown_shape().as_proto(),
/external/llvm-project/mlir/test/Dialect/Shape/
Dops.mlir30 %0 = "shape.unknown_shape"() : () -> !shape.shape

12