Searched defs:xla_shape (Results 1 – 10 of 10) 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()
393 xla::Shape xla_shape; in ReadVariableInputTensor() local531 xla::Shape xla_shape; in AssignVariableTensor() local
313 xla::Shape xla_shape; in TEST_F() local1036 xla::Shape xla_shape; in TEST_F() local1074 xla::Shape xla_shape; in TEST_F() local
453 xla::Shape xla_shape; in XlaCompiler() local
87 xla::Shape xla_shape; in Compile() local137 xla::Shape xla_shape; in Compile() local174 xla::Shape xla_shape; in Compile() local216 xla::Shape xla_shape; in Compile() local
47 xla::Shape xla_shape; in Compile() local215 xla::Shape xla_shape; in Compile() local250 xla::Shape xla_shape; in Compile() local278 xla::Shape xla_shape; in Compile() local
106 xla::Shape xla_shape = xla::ShapeUtil::MakeShapeWithDescendingLayout( in Compile() local
259 xla::Shape xla_shape; in Compile() local306 xla::Shape xla_shape; in Compile() local339 xla::Shape xla_shape; in Compile() local
92 xla::Shape xla_shape; in XlaDeviceContext() local
230 xla::Shape xla_shape; in XRTAllocateFromTensorOp() local