Searched defs:device_shape (Results 1 – 11 of 11) sorted by relevance
59 Shape* device_shape) { in ReadDynamicShapes()
275 ShapedBuffer* device_buffer, Shape* device_shape, in ReadDynamicShapesOnCpu()
112 xla::Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape() local
62 xla::Shape device_shape = ApiConverter::FromC(&c_device_shape); in HostShapeToDeviceShape() local328 xla::Shape* device_shape) { in ReadDynamicShapes()
74 Shape device_shape = shape; in AllocateOutputMemoryWithInputReuse() local
417 xla::Shape* device_shape) { in ExpandTreeOfTuples()463 xla::Shape device_shape; in MakeTuple() local
279 const xla::Shape& device_shape = in UpdateOutputVariables() local
459 const xla::Shape& device_shape = in AllocateOutputTensors() local
205 Shape* device_shape) { in ReadDynamicShapes()
249 const xla::Shape& device_shape = in PopulateInputs() local