Searched defs:xla_shape (Results 1 – 11 of 11) sorted by relevance
27 xla::Shape xla_shape; in HostTensorToBorrowingLiteral() local43 xla::Shape xla_shape; in HostTensorToMutableBorrowingLiteral() local50 const xla::Shape& xla_shape, Tensor* host_tensor, in HostTensorToMutableBorrowingLiteral()
322 const xla::Shape& xla_shape = resource->representation_shape().value(); in BuildComputation() local544 xla::Shape xla_shape; in XlaCompiler() local709 xla::Shape xla_shape = absl::get<xla::Shape>(args[i].shape); in CompileFunction() local769 xla::Shape* xla_shape) { in XLAShapeForArgument()
420 xla::Shape xla_shape; in ReadVariableInputTensor() local558 xla::Shape xla_shape; in AssignVariableTensor() local
310 xla::Shape xla_shape; in TEST_F() local354 xla::Shape xla_shape; in TEST_F() local407 xla::Shape xla_shape; in TEST_F() local1091 xla::Shape xla_shape; in TEST_F() local1130 xla::Shape xla_shape; in TEST_F() local
48 xla::Shape xla_shape; in Compile() local225 xla::Shape xla_shape; in Compile() local264 xla::Shape xla_shape; in Compile() local292 xla::Shape xla_shape; in Compile() local325 xla::Shape xla_shape; in Compile() local
124 xla::Shape xla_shape; in Compile() local179 xla::Shape xla_shape; in Compile() local218 xla::Shape xla_shape; in Compile() local273 xla::Shape xla_shape; in Compile() local
209 xla::Shape xla_shape; in Compile() local253 xla::Shape xla_shape; in Compile() local293 xla::Shape xla_shape; in Compile() local338 xla::Shape xla_shape; in Compile() local371 xla::Shape xla_shape; in Compile() local
34 xla::Shape xla_shape; in TestShapeRepresentation() local
78 xla::Shape& xla_shape = individual_arg_shapes.back(); in GetXlaInputShapes() local
97 xla::Shape xla_shape; in XlaDeviceContext() local
278 xla::Shape xla_shape; in XRTAllocateFromTensorOp() local