Searched refs:shape_representation_fn_ (Results 1 – 4 of 4) sorted by relevance
85 shape_representation_fn_(std::move(shape_representation_fn)), in XlaDeviceContext()89 if (!shape_representation_fn_) { in XlaDeviceContext()90 shape_representation_fn_ = [](const TensorShape& shape, in XlaDeviceContext()130 shape_representation_fn_(device_tensor->shape(), in CopyCPUTensorToDevice()
73 return shape_representation_fn_; in shape_representation_fn()83 XlaCompiler::ShapeRepresentationFn shape_representation_fn_; variable226 const XlaCompiler::ShapeRepresentationFn shape_representation_fn_; variable
83 return shape_representation_fn_; in shape_representation_fn()111 XlaCompiler::ShapeRepresentationFn shape_representation_fn_; variable
147 shape_representation_fn_(std::move(shape_representation_fn)), in Metadata()205 shape_representation_fn_(options.shape_representation_fn), in XlaDevice()327 client(), shape_representation_fn_, thread_pool_.get()); in GetDeviceContextLocked()