Searched defs:static_shape (Results 1 – 5 of 5) sorted by relevance
58 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape): argument
455 PartialTensorShape static_shape(arg.shape()); in ComputeArgumentShapes() local
1312 def _is_scalar_helper(self, static_shape, dynamic_shape_fn): argument
1348 const Shape static_shape = xla::ShapeUtil::MakeStaticShape(current_shape); in GetOrCreateArrayView() local
2696 static Py_ssize_t static_shape[1] = {12}; variable