Searched defs:static_shape (Results 1 – 8 of 8) sorted by relevance
58 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape): argument
461 PartialTensorShape static_shape(arg.shape()); in ComputeArgumentShapes() local
1391 const Shape static_shape = xla::ShapeUtil::MakeStaticShape(current_shape); in GetOrCreateArrayView() local1595 const Shape static_shape = in Initialize() local
1312 def _is_scalar_helper(self, static_shape, dynamic_shape_fn): argument
149 auto static_shape = shaped_type.getShape(); in GetShape() local
1422 Shape static_shape = operand->shape(); in RewriteDynamicBinaryOp() local
2371 auto static_shape = ShapeUtil::MakeShape(S32, {4}, {false}); in TEST_F() local
2696 static Py_ssize_t static_shape[1] = {12}; variable