Searched defs:static_shape (Results 1 – 9 of 9) sorted by relevance
251 TensorShape static_shape; in Compile() local435 TensorShape static_shape; in Compile() local
54 def _compare(self, x_in, y_in, adjoint_a, adjoint_b, static_shape): argument
461 PartialTensorShape static_shape(arg.shape()); in ComputeArgumentShapes() local
134 auto static_shape = shaped_type.getShape(); in GetShape() local
1305 def _is_scalar_helper(self, static_shape, dynamic_shape_fn): argument
1447 const Shape static_shape = xla::ShapeUtil::MakeStaticShape(current_shape); in GetOrCreateArrayView() local
1467 Shape static_shape = operand->shape(); in RewriteDynamicBinaryOp() local
2385 auto static_shape = ShapeUtil::MakeShape(S32, {4}, {false}); in TEST_F() local
2696 static Py_ssize_t static_shape[1] = {12}; variable